python - 我该如何解释这个符号？

Question

符号s[s]如何以及为什么起作用？

我正在学习 kaggle.com 上的一门微类(class)，它们使用符号 s[s]，如下所示。我以前没见过。X_train 是一个 pandas DataFrame。

它是一个列表自身切片吗？有人能帮忙澄清一下吗？

s = (X_train.dtypes == 'object') ## assigns True to variables == 'object'
object_cols = list(s[s].index)

> s

Type              True
Method            True
Regionname        True
Rooms            False
Distance         False
Postcode         False
Bedroom2         False
Bathroom         False
Landsize         False
Lattitude        False
Longtitude       False
Propertycount    False
dtype: bool

> s[s]

Type          True
Method        True
Regionname    True
dtype: bool

Answer 1

这相当复杂。

X_train 是 pandas 数据框。

X_train.dtypes 返回一个 pandas Series，其中索引(每行的名称)等于列名称。

我们现在对 Series 执行 == 操作，返回一个新系列，其值为 true 或 false。所以它看起来像:

一个真
b 错误
c 正确

现在我们到达 x[x]，它表示删除“false”值，给出一个新的系列:

一个真
c 正确

现在我们做.index并将其转换成一个列表来给出

['a', 'c']