Манипулировать строками на основе условия

Question

Я хочу манипулировать строками панд в зависимости от условий.Например, я хочу перезаписать каждое поле в method строкой 'BBBB', если поле в type является пустым NaN:

newcolumn = []
for index, row in results_DF.iterrows():
    newcolumn.append('BBBB' if row['type'] is np.nan else row['method'])
results_DF['method'] = pd.Series(newcolumn)

Эта реализация выглядит довольно некрасиво.Как написать это лучше - в более функциональном стиле?

Answer 1

попробуйте это,

mask=results_DF['type'].isnull()
results_DF.loc[mask]='BBBB'

Answer 2

Использование DataFrame.loc с логическим значением amsk, созданным Series.isna:

results_DF.loc[results_DF['type'].isna(), 'method'] = 'BBBB'

#oldier pandas versions
#results_DF.loc[results_DF['type'].isnull(), 'method'] = 'BBBB'

Другое решение с numpy.where:

results_DF['method'] = np.where(results_DF['type'].isna(), 'BBBB', results_DF['method'])

Или решение от @Jon Clements, спасибо:

results_DF['method'] = results_DF.where(results_DF['type'].notnull(), 'BBBB')

Sample

results_DF = pd.DataFrame({'method': ['a','s','d'],
                           'type':[np.nan, np.nan, 4]})

print (results_DF)
  method  type
0      a   NaN
1      s   NaN
2      d   4.0

results_DF.loc[results_DF['type'].isna(), 'method'] = 'BBBB'
print (results_DF)
  method  type
0   BBBB   NaN
1   BBBB   NaN
2      d   4.0