如何创建一个列表作为字典的键，并将其添加到不同的明细表中？

Question

我有一个for循环，它运行在一个CSV文件中，获取某些元素，并基于两个变量创建一个字典

代码：

for ind, row in sf1.iterrows():
    sf1_date = row['datekey']
    sf1_ticker = row['ticker']
    company_date[sf1_ticker] = [sf1_date]

例如，在for循环的第一次迭代中，sf1_ticker = 'AAPL'和sf1_date = '2020/03/01'，下一次，sf1_ticker = 'AAPL'和sf1_date = '2020/06/01'，我如何使字典中的'AAPL'键等于['2020/03/01', '2020/06/01']

Answer 1

您可以使用defaultdict，它可以设置为向任何不存在的项添加空列表。否则，它的作用通常类似于词典

from collections import defaultdict

rows = [
    ['AAPL', '2020/03/01'], 
    ['AAPL', '2020/06/01'],
    ['GOOGL', '2021/01/01']
]

company_date = defaultdict(list)
for ticker, date in rows:
    company_date[ticker].append(date)

print(company_date)
# defaultdict(<class 'list'>, {'AAPL': ['2020/03/01', '2020/06/01'], 'GOOGL': ['2021/01/01']})