系列是什么意思[系列]

Question

我目前正在学习一些机器学习，并使用 pandas，在其中一个示例中他们这样做了

# Get list of categorical variables
s = (X_train.dtypes == 'object')
object_cols = list(s[s].index)

print("Categorical variables:")
print(object_cols)

我目前正在努力理解他们做 s[s].index 时发生了什么，s[s] 怎么可能？有人可以帮我吗？ 提前致谢

这是我正在阅读的链接：https://www.kaggle.com/alexisbcook/categorical-variables

Answer 1

您可以按条件索引系列，例如myseries[myseries == 4] 将返回等于 4 的元素。在你的情况下，如果你有一个布尔值，这两个表达式是相同的：

myseries[myseries == True] 
myseries[myseries] 

因此，在您的示例中，您返回 True 的所有元素的索引：

myseries = pd.Series([True, False, True, False])
x = myseries[myseries] # returns Series: (0, True) (2, True)
x = x.index # returns [0,2]

在您的 kaggle 示例的上下文中，这些布尔值由 s = (X_train.dtypes == 'object') 构造，即您发布的代码 sn-p 获取类型为 object 的每个训练示例的索引。

我个人不会推荐这个快捷方式，因为它会不必要地模糊代码并且编写 s[s==True] 并不是太多工作。