從數據框中的列中選擇特定值

Question

我有一個只有兩列的數據集。 我想根據某一條件從其中提取一小部分。 將此視為我的數據集。

A    B
1    10
1    9
2    11
3    12
3    11
4    9

假設我只想提取那些B的值從10-12的行，那么我將得到一個新的數據集，如下所示：

A    B
1    10
2    11
3    12
3    11

我嘗試使用df.loc [df [“ B”] == range（10，12）]，但是它不起作用，有人可以幫助我嗎？

Answer 1

您可以使用.between

In [1031]: df.loc[df.B.between(10, 12)]
Out[1031]:
   A   B
0  1  10
2  2  11
3  3  12
4  3  11

或者， isin

In [1032]: df.loc[df.B.isin(range(10, 13))]
Out[1032]:
   A   B
0  1  10
2  2  11
3  3  12
4  3  11

或者， query

In [1033]: df.query('10 <= B <= 12')
Out[1033]:
   A   B
0  1  10
2  2  11
3  3  12
4  3  11

或者，good'ol boolean

In [1034]: df.loc[(df.B >= 10) & (df.B <= 12)]
Out[1034]:
   A   B
0  1  10
2  2  11
3  3  12
4  3  11

Answer 2

這是另一個（不使用.loc()或.query() ），它看起來更像是最初的（失敗的）嘗試：

df[df.B.isin(range(10,13))]