python - 数据框中的每个单元格都包含一个值列表 - 如何选择第一个值？

Question

我有一个数据框，我想选择每个单元格列表中的第一个值。

我已经尝试过:

for i in clean_columns:
    print(clean_columns[i][:][0][0])

但它只选择第一行。如何从每一行中选择第一个值，以便留下一个数据框减去括号中的值。谢谢

Answer 1

您可以使用applymap :

df.applymap(lambda x: x[0])

或者，stack ，使用str定位器，unstack :

df.stack().str[0].unstack()

或者，apply与 str :

df.apply(lambda c: c.str[0])

示例:

# input
df = pd.DataFrame([[[0, 1] for _ in range(3)] for _ in range(3)])
#         0       1       2
# 0  [0, 1]  [0, 1]  [0, 1]
# 1  [0, 1]  [0, 1]  [0, 1]
# 2  [0, 1]  [0, 1]  [0, 1]

# output
   0  1  2
0  0  0  0
1  0  0  0
2  0  0  0