팬더 : 목록 값 비교 및 ​​새 열 쓰기

Question

내 my old post과 관련된 새로운 질문이 있습니다. 그 게시물에서 문제는 단순화되었고, 비교할 2 개의 w 만있었습니다. 이제 주파수가 (1,1,0) 인 2 이상, 예를 들어 3이 있다고 가정합니다. 이 목록을 확인하고 싶습니다. 최대 값이 인 경우 을 두 번 표시하고 0을 쓰고 그렇지 않으면 최대 값의 열 레이블을 쓰십시오. 그 게시물의 첫 번째 답변을 적용하려했지만 막혔습니다.

도움을 주시면 감사하겠습니다. 관심을 가져 주셔서 감사합니다. :-)

편집 :

는

내 진짜 DF의 예는이 경우

|FID |geometry|GridCode|catarro|constipado|gripa|gripe|resfriado|resfrio 0 |9592|... |9592 |1 |0 |0 |3 | 3 | 1

는 원하는 출력 될 것이다 : |FID |geometry|GridCode|catarro|constipado|gripa|gripe|resfriado|resfrio| max 0 |9592|... |9592 |1 |0 |0 |3 | 3 | 1 | 0

Answer 1

편집 :

을

# You should drop all extra fields 
# don't worry they are still present in original dataframe (df) 
words = df.drop(['FID'], axis=1) 

# Get maximums for each row 
maxes = words.max(axis=1) 

# Create new column with the features names with maximum values 
df['max'] = words.idxmax(axis=1) 

# Create a mask with non-accepted rows 
mask = (
    words.values.ravel() == maxes.values.repeat(len(words.columns)).ravel() 
).reshape(-1,len(words.columns)).astype(int).sum(axis=1)>1 

# Wipe 'max' column in non-accepted rows 
df.ix[mask,'max'] = 0