簡體   English   中英

從 csv 創建嵌套字典

[英]Create nested dictionary from csv

我在 csv 中有一些數據,打開時看起來像這樣:

Example Data
['', 'Name', 'Phone', 'Address', 'City', 'Country', 'Email']
['1', 'Bob Myers', '410-504-5887', '12334 Hamilton Way', 'Toronto', 'Canada', 'bob@gmail.com']
['2', 'Carlton James', '455-323-8479', '1234 James Rd', 'New York', 'USA', 'carlton@example.com']
['3', 'Frank Wright', '744-521-9874', '567 Travis St', 'Boston', 'USA', 'fw4322@yahoo.com']

我想創建一個嵌套字典,其中我在 '' 中取整數並將其用作每個嵌套字典的鍵。 我發現了一些類似的問題並嘗試過:

import csv
f = csv.DictReader(open('data.csv'))

result = {}
for row in f:
    key = row.pop('')
    result[key] = row
print(result)

這產生了類似的東西:

{'1': {'Name': 'Bob Myers', 'Phone': '410-504-5887', 'Address': '12334 Hamilton Way', 'City': 'Toronto',
'Country': 'Canada', 'Email': 'bob@gmail.com'}...

我在我的代碼中做了什么改變來讓我的數據看起來像這樣(沒有熊貓):

my_dict = {{'Name': {1: 'Bob Myers', 2: 'Carlton James', 3: 'Frank Wright'}}, 
 {'Phone': {1: '410-504-5887', 2: '455-323-8479', 3: '744-521-9874'}}, 
 {'Address': {1: '12334 Hamilton Way', 2: '1234 James Rd', 3: '567 Travis St'}},
 {'City': {1: 'Toronto', 2: 'New York', 3: 'Boston'}},
 {'Country': {1: 'Canada', 2: 'USA', 3: 'USA'}},
 {'Email': {1: 'bob@gmail.com', 2: 'carlton@example.com', 3: 'fw4322@yahoo.com'}}
}

最好的方法是(ab)使用 Pandas ;-)

import pandas as pd
d = pd.DataFrame([result]).to_dict()
print(d)

對於測試用例result

{'1': {'Name': 'Bob Myers', 'Phone': '410-504-5887', 'Address': '12334 Hamilton Way', 'City': 'Toronto',
'Country': 'Canada', 'Email': 'bob@gmail.com'}

輸出是這樣的:

{'Name': {0: 'Bob Myers'}, 'Phone': {0: '410-504-5887'}, 'Address': {0: '12334 Hamilton Way'}, 'City': {0: 'Toronto'}, 'Country': {0: 'Canada'}, 'Email': {0: 'bob@gmail.com'}} 

由於您有一個 csv 數據文件,因此最好使用pandas

import pandas as pd

a = pd.read_csv(file location)

nested_dict = a.todict()

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM