Python 字母数字和字符比较

Question

我有一个 CSV 数据集，我正在尝试选择“INCOME”列等于“50K”的所有行

这是我所拥有的，但它只返回没有任何行的列名。

df.loc[df.INCOME == '>50K',:]

我的代码基本上说选择“df”数据集的 INCOME 列等于 '>50K' 的所有行

有什么帮助吗？

Answer 1

你可以用

df.INCOME=df.INCOME.str.strip()
df.loc[df.INCOME == '>50K',:]

Answer 2

@Wen 已经展示了一个很好的答案，但是您可以尝试使用元素运算符的替代方法，就像您最初在 POST 中希望的那样。

示例数据框：

>>> df
   NUM INCOME  CAP
0    1    20k    5
1    2    30k    9
2    8    50k    2
3    3    40k    7
4    6    50k    7
5    4    80k    3
6    8    20k    8
7    9    90k   10
8    6    60k    4
9   10    10k    7

使用逐元素逻辑OR(|)：

>>> df.loc[ (df['INCOME'] <='10k') | (df['INCOME'] >='50k') ]
       NUM INCOME  CAP
    2    8    50k    2
    4    6    50k    7
    5    4    80k    3
    7    9    90k   10
    8    6    60k    4
    9   10    10k    7

或者：

>>> df.loc[ (df['INCOME'] <='20k') | (df['INCOME'] =='50k') ]
   NUM INCOME  CAP
0    1    20k    5
2    8    50k    2
4    6    50k    7
6    8    20k    8
9   10    10k    7