pandas 中的列表理解

Question

我的列表理解有问题。我编写了如下代码来获取包含 nan 的列表，其中我的 df 的第二列中的值为 True

[np.nan for name in data_col.iloc[:,0] if data_col.iloc[:,1]==True]

我收到了错误，而不是预期的输出：

ValueError: The truth value of a Series is ambiguous. Use a.empty, a.bool(), a.item(), a.any() or a.all().

有人可以帮忙吗？

Answer 1

在您的 if 语句中，每次循环并执行您的 if 语句时，您都会返回一个完整的列表，您需要访问单个元素for x in range(len(data_col.iloc[:,0])):    if data_col.iloc[x,1]:        np.nan或者这应该工作我通常不写一个衬里。[np.nan for x in range(len((data_col.iloc[:,0])) if data_col.iloc[x,1]]