Series 的真值不明确（如果）

Question

当我想添加新列时，if 语句出现问题。

import pandas as pd
scan = pd.DataFrame([[1,2,3],['a','b','c']], columns=['st','nd','rd'])
scan['th'] = 0 if scan['st'] == 0 else 1

ValueError：Series 的真值不明确。使用 a.empty、a.bool()、a.item()、a.any() 或 a.all()。

我怎样才能解决这个问题？

Answer 1

使用 numpy.where 将这些条件和值应用于整个数组：

scan['th'] = np.where(scan['st'] == 0, 0, 1)

将前一种方法与这种方法进行基准测试可能会很有趣：

scan['th'] = (scan['st'] != 0).astype(int)