python - 访问数据框内的字典

Question

我有以下数据框(有 1324 行): enter image description here

我需要找出哪些城市适合外卖(属性字典中'Take_out': True)

Answer 1

为了得到这个答案，我首先创建了一个虚拟 DataFrame 进行测试:

import numpy as np
import pandas as pd
# create a dictionary list
d = list({'Take-out': True} for x in np.arange(10))
ddf = pd.Series(d, name='attributes')
ddf = pd.DataFrame(ddf)
ddf.index.name = 'cities'
print(ddf)

这会给出与图像中类似的 DataFrame。

接下来，迭代 DataFrame，访问“属性”列，如下所示:

# cities buffer will hold successes
cities = []
# iterate over the list of dictionaries:
for i, each in enumerate(ddf['attributes']):
    # check if the keys is in that dictionary, if so, keep the city name
    if 'Take-out' in ddf['attributes'][i].keys():
        # the index is named 'cities' and each position is a city name, so:
        cities.append(ddf.index[i])
print(cities)