DataFrame选取行数据

选取某列相同数值的行数据

import pandas as pd
myList = [[1, 10, 100],
	  [2, 20, 200],
	  [3, 30, 300],
	  [1, 40, 400],
      [1, 50, 500]]
df1 = pd.DataFrame(data = myList)
print(df1)

df2 = pd.DataFrame(myList,
		           index = ['one', 'two', 'three', 'four','five'],
		           columns = ['time', 'single', 'double'])
print(df2)

df3=df2.loc[df2['time'] == 1]  #选取time列数据为1的所有行
print(df3)

list=[1,2]
df4=df2.loc[df2['time'] .isin(list)] #选取time列包含在list内的数据
print(df4)

【df1：】

     0   1    2
0   1  10  100
1   2  20  200
2   3  30  300
3   1  40  400
4   1  50  500

【df2：】

          time  single  double
one       1      10     100
two        2      20     200
three     3      30     300
four       1      40     400
five        1      50     500

【df3：】

        time  single  double
one      1      10     100
four     1      40     400
five      1      50     500

【df4：】 

          time  single  double
one      1      10     100
two      2      20     200
four     1      40     400
five      1      50     500