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]]