如何使用元组查询熊猫列

Question

如何查询包含元组的列？

     # Tuples in a dataframe
     _df = pd.DataFrame({'C': [(0,1), (2,3), (4,5)]})
     display(_df)

     for i in range(len(_df)):
         _df.loc[i, 'D'] = sum(_df.loc[i, 'C'])
     display(_df)


          C
    0   (0, 1)
    1   (2, 3)
    2   (4, 5)

例如，如何查询 C 中具有第一个元素 == 2 的元组的所有元组？

    _df[_df['C'] == (2, 3)]

匹配元组。但是如何只匹配元组的一个元素？

Answer 1

使用str[0]元组的选择第一值：

print (_df['C'].str[0])
0    0
1    2
2    4
Name: C, dtype: int64

print (_df[_df['C'].str[0] == 2])
        C
1  (2, 3)