[pandas] 選取同欄位的不同資料

上大神的課總是可以學到很多神秘的招數XD


又學了一招 "選取同欄位的不同資料"

import pandas as pd



#建立 DataFrame

a = [1,2,3,4,5]

b = ['a','b','c','d','e']

c = zip(a,b)

data = pd.DataFrame(c, columns =['no.', 'value'])

data.head()

#長得像這樣

# no. value

# 1 a

# 2 b

# 3 c

# 4 d

# 5 e



chosen_data = data[ (data['value']=='e') | (data['value']=='b') ]

chosen_data.head()

#最後挑出來的

# no. value

# 2 b

# 5 e