pandas DataFrame selection by element in array

前端 未结 1 411
孤街浪徒
孤街浪徒 2021-01-21 17:15

Trying to select subset of df based on the occurrence of an element in an array in the df.

df = pd.DataFrame()
vals = []
for i in range(3):
    vals.append(np.li         


        
相关标签:
1条回答
  • 2021-01-21 17:39

    You need apply with in for boolean mask, if need filter use boolean indexing:

    print (df.vals.apply(lambda x: 0.5 in x))
    0    False
    1    False
    2     True
    Name: vals, dtype: bool
    
    print (df[df.vals.apply(lambda x: 0.5 in x)])
                  vals
    2  [0.0, 0.5, 1.0]
    
    0 讨论(0)
提交回复
热议问题