python - 当单元格中的值是列表时，Pandas 数据框条件

Question

如果同一行中B列的值是列表[0，我如何找到列x中单元格的值,1]?

A 和 B 列中的每个单元格都是整数列表。

               x      y       A       B 
0           ding   ding    [0, 1]  [0, 0] 
1           ding   dong    [0, 1]  [0, 1] 
2           ding   ding    [0, 1]  [0, 0] 
3           ding   ding    [0, 1]  [0, 0]  
4           ding   dong    [0, 1]  [0, 1] 
5           ding   ding    [0, 1]  [0, 0] 
6           ding   ding    [0, 1]  [0, 0] 

Answer 1

IIUC 您可以创建与 DataFrame 具有相同索引的 Series 并将其与列 B 进行比较:

s = pd.Series([[0, 1]], index = df.index)
print s
0    [0, 1]
1    [0, 1]
2    [0, 1]
3    [0, 1]
4    [0, 1]
5    [0, 1]
6    [0, 1]
dtype: object

print df['B'] == s
0    False
1     True
2    False
3    False
4     True
5    False
6    False
dtype: bool

print df[df['B'] == s]
      x     y       A       B
1  ding  dong  [0, 1]  [0, 1]
4  ding  dong  [0, 1]  [0, 1]

print df.x[df['B'] == s]
1    ding
4    ding
Name: x, dtype: object