我有一个包含列表列的数据框:
col_1
[A, A, A, B, C]
[D, B, C]
[C]
[A, A, A]
NaN
我想创建一个新列,如果列表以3 *
A
开头,则返回1,否则返回0:col_1 new_col
[A, A, A, B, C] 1
[D, B, C] 0
[C] 0
[A, A, A] 1
NaN 0
我尝试过这个,但是没有成功:
df['new_col'] = df.loc[df.col_1[0:3] == [A, A, A]]
df['col_1'] = df['col_1'].str.strip('[]').str.split(',')
。 - jezrael