在一个元素的数组上调用list（）方法会引发TypeError：在0维数组上的迭代

Posted 2021-04-19

在熊猫数据帧单行上的

calling list（）方法会引发错误。例如，

 d = {'col1': ['a', 'b', 'a'], 'col2': ['c', 'd', 'e']}
 df = pd.DataFrame(data=d)
 df

现在，

list(df.loc[df.col1 == 'a']['col2'].values.squeeze())

很好

但是

list(df.loc[df.col1 == 'b']['col2'].values.squeeze())

提高：

TypeError: iteration over a 0-d array

如何解决此问题？

答案

您可以在这里使用pd.Series.tolist()。

pd.Series.tolist()