python - Pandas :从列列表中替换值列表

Question

我有很多行、很多列的数据框，它们具有需要替换的不同“占位符”值(在列的子集中)。我已经在论坛中阅读了很多使用嵌套列表或字典的示例，但没有运气变化..

# A test dataframe
df = pd.DataFrame({'Sample':['alpha','beta','gamma','delta','epsilon'],
                  'element1':[1,-0.01,-5000,1,-2000], 
                  'element2':[1,1,1,-5000,2], 
                  'element3':[-5000,1,1,-0.02,2]})

# List of headings containing values to replace
headings = ['element1', 'element2', 'element3']

我正在尝试做这样的事情(显然这行不通):

 # If any rows have value <-1, NaN
 df[headings].replace(df[headings < -1], np.nan)

 # If a value is between -1 and 0, make a replacement
 df[headings].replace(df[headings < 0 & headings > -1], 0.05)

那么，是否有更好的方法可以使用循环或花哨的 pandas 技巧来完成此任务？

Answer 1

您可以将Sample列设置为索引，然后根据条件替换整个数据框上的值:

df = df.set_index('Sample')
df[df < -1] = np.nan
df[(df < 0) & (df > -1)] = 0.05

给出:

#           element1    element2    element3
#  Sample           
#   alpha       1.00        1.0          NaN
#    beta       0.05        1.0         1.00
#   gamma        NaN        1.0         1.00
#   delta       1.00        NaN         0.05
# epsilon        NaN        2.0         2.00