将列表传递给Pandas loc方法

时间:2020-07-24 19:19:21

标签: pandas dataframe loc

我想更改熊猫数据框中某些列的值。但是,如果我在loc内传递列列表,我似乎无法做。

df = pd.DataFrame({
"ID" : [1, 2, 3, 4, 5],
"QA_needed" : [0, 1, 1, 0, 1],
"QC_needed" : [1, 0, 1, 0, 0],
"Report_needed" : [1, 1, 1, 0, 1]
})

df.loc[:, ["QA_needed", "Report_needed"]].replace({1: "True", 0: "False"}, inplace=True)

为此,我必须分别替换每一列的值

df.loc[:, "QA_needed"].replace({1: "True", 0: "False"}, inplace=True)
df.loc[:, "QC_needed"].replace({1: "True", 0: "False"}, inplace=True)

是否可以将列表["QA_needed", "Report_needed"]传递给loc方法?

2 个答案:

答案 0 :(得分:3)

尝试update

df.update(df.loc[:, ["QA_needed", "Report_needed"]].replace({1: "True", 0: "False"}))
df
Out[96]: 
   ID QA_needed  QC_needed Report_needed
0   1     False          1          True
1   2      True          0          True
2   3      True          1          True
3   4     False          0         False
4   5      True          0          True

答案 1 :(得分:2)

您还可以分配一个布尔值(bool):

df = df.assign(**df[['QA_needed','Report_needed']].astype(bool))

   ID  QA_needed  QC_needed  Report_needed
0   1      False          1           True
1   2       True          0           True
2   3       True          1           True
3   4      False          0          False
4   5       True          0           True