根据列表值选择Pandas行

时间:2013-08-27 13:30:00

标签: python pandas

我有一个看起来像这样的Pandas DataFrame:

<class 'pandas.core.frame.DataFrame'>
Int64Index: 1072 entries, 0 to 1071
Data columns (total 10 columns):
city            1072  non-null values
latitude        1072  non-null values
longitude       1072  non-null values
manufacturer    1072  non-null values
name            1072  non-null values
offering        1072  non-null values
platformID      1072  non-null values
procedure       1072  non-null values
properties      1072  non-null values
sensorID        1072  non-null values
dtypes: object(10)

properties是一个字符串值列表:

df_devices.head(1).properties
Out[79]: 0    [urn:average_wind, urn:dew_point]

我想选择仅包含'urn:dew_point'条目的记录,但我不确定如何过滤它们(使用isin或其他方式)

1 个答案:

答案 0 :(得分:3)

您只需使用apply即可:

In [11]: df = pd.DataFrame([[['urn:dew_point'], 1]], columns=['properties', 'id'])

In [12]: df
Out[12]:
        properties  id
0  [urn:dew_point]   1

In [13]: df[df['properties'].apply(lambda x: 'urn:dew_point' in x)]
Out[13]:
        properties  id
0  [urn:dew_point]   1

如果这只是字符串列的一部分,您可以使用str.contains

In [21]: df = pd.DataFrame([['urn:dew_point', 1]], columns=['properties', 'id'])

In [22]: df['properties'].str.contains('urn:dew_point')
Out[22]:
0    True
Name: properties, dtype: bool

In [23]: df[df['properties'].str.contains('urn:dew_point')]
Out[23]:
      properties  id
0  urn:dew_point   1