python - 如果另一列数据在列表中， Pandas 会更改一列数据

Question

我在这样的数据框中有一些数据:

COLUMNS Point,x,y,z,Description,Layer,Date
1,224939.203,1243008.651,1326.774,F,C-GRAD-FILL,09/22/18 07:24:34,
1,225994.242,1243021.426,1301.772,BS,C-GRAD-FILL,09/24/18 08:24:18,
451,225530.332,1243016.186,1316.173,GRD,C-TOE,10/02/18 11:49:13,
452,225522.429,1242996.017,1319.168,GRD,C-TOE KEY,10/02/18 11:49:46,

我想尝试检查一个字符串列表以查看它是否匹配，然后更改另一个列的值。

myList = ["BS", "C"]
if df['Description'].isin(myList)) == True:
     df['Layer']="INLIST"
     df['Date']="1/1/2018'

Answer 1

使用掩码 pd.DataFrame.loc :

mask = df['Description'].isin(['BS', 'C'])
df.loc[mask, 'Layer'] = 'INLIST'
df.loc[mask, 'Date'] = '1/1/2018'

或者，使用 pd.Series.mask :

mask = df['Description'].isin(['BS', 'C'])
df['Layer'].mask(mask, 'INLIST', inplace=True)
df['Date'].mask(mask, '1/1/2018', inplace=True)

请注意，您可能应该将日期存储为 datetime 对象，例如pd.Timestamp('2018-01-01')，或通过 pd.to_datetime 转换您的系列.通常不需要 Python 级别的循环。