以下是一个示例代码,展示了如何遍历Pandas数据框并将值追加到一个列表中,该列表是一个字典中以条件值为键的值:
import pandas as pd
# 创建一个示例数据框
data = {'Name': ['Tom', 'Nick', 'John', 'Tom'],
'Age': [20, 21, 19, 22],
'City': ['New York', 'Paris', 'London', 'Tokyo']}
df = pd.DataFrame(data)
# 创建一个空字典,以条件值为键,值为一个空列表
result = {}
# 遍历数据框的每一行
for index, row in df.iterrows():
# 获取条件值
condition = row['Name']
# 将值追加到对应的列表中
if condition in result:
result[condition].append(row['Age'])
else:
result[condition] = [row['Age']]
# 打印结果
print(result)
这段代码首先创建了一个示例数据框,然后创建了一个空字典result
。接下来,使用iterrows()
方法遍历数据框的每一行。在每一行中,获取条件值(这里是Name
列的值),然后将该行的Age
值追加到对应的列表中。如果条件值在字典中不存在,则创建一个新的键,并初始化一个空列表。最后,打印结果字典。