在 Python 中迭代嵌套字典

Question

我的字典結構如下：

stockData = {
    'AAPL': {
        'beta': 1.01833975315094,
        'company_name': 'Apple',
        'dividend': 1.9341673320912078, 
        'total':300}, 
    'GOOG': {
        'beta': 1.01833975315094,
        'company_name': 'Apple',
        'dividend': 1.9341673320912078, 
        'total':300}
     }

這是我分配值的區域：

for row in range(2, sheet.max_row + 1):
    # Each row in the spreadsheet has data for one census tract.
    company_name  = sheet['A' + str(row)].value
    ticker = sheet['B' + str(row)].value
    sector = sheet['C' + str(row)].value
    shares = sheet['D' + str(row)].value
    price = sheet['E' + str(row)].value
    total = sheet['F' + str(row)].value
    beta = sheet['G' + str(row)].value
    dividend = sheet['H' + str(row)].value
    number_stocks+=1

    # Make sure the key for this ticker exists.
    stockData.setdefault(ticker,{})
    stockData[ticker]['company_name'] = company_name
    stockData[ticker]['sector'] = sector
    stockData[ticker]['shares'] = shares
    stockData[ticker]['price'] = price
    stockData[ticker]['total'] = total
    stockData[ticker]['dividend'] = dividend
    stockData[ticker]['beta'] = beta

我遇到了一個問題，我在字典上使用 for 循環進行迭代以添加一個新值“百分比”，即“總數”除以所有股票的所有總數的總和。

def get_portfolio_value(stocks,item):
    portValue = 0
    percentage = 0
    for k, v in stocks.items():
        portValue = portValue + v.get(item,0)
        stockData[ticker]['percentage'] = stockData[ticker]['total']/portValue
    print(portValue)

get_portfolio_value(stockData,'total')

問題是 get_portfolio_value 函數讓我獲得了投資組合的整個 portValue 總數，但是我試圖將投資組合的百分比添加到每只股票的行不起作用 - 它只是奇怪地出現僅股票之一。 有人可以建議嗎？

Answer 1

這就是你想要的：

stock_data = {
    'AAPL': {'beta': 1.01833975315094, 'company_name': 'Apple', 'dividend': 1.9341673320912078, 'total':300},
    'GOOG': {'beta': 1.01833975315094, 'company_name': 'Apple', 'dividend': 1.9341673320912078, 'total':300}
}

stock_sum = sum(stock_data[item]['total'] for item in stock_data)

for item in stock_data:
    stock_data[item]['percentage'] = int((float(stock_data[item]['total']) / stock_sum) * 100)

結果：

>>> for item in stock_data:
...     print stock_data[item]['percentage']
...
50
50

Answer 2

如果我正確理解你的問題，你可以嘗試如下（我只是復制了下面的stockData ）：

stockData = {
    'AAPL': {
        'beta': 1.01833975315094,
        'company_name': 'Apple',
        'dividend': 1.9341673320912078, 'total': 300},
    'GOOG': {
        'beta': 1.01833975315094,
        'company_name': 'Apple',
        'dividend': 1.9341673320912078, 'total': 300}}

我想這就是你的意思：

def get_portfolio_value(stocks, item='total'):
     portValue = 0
     for v in stocks.values():
        portValue += v[item]
     for k in stocks.keys():
        stocks[k].update({'percentage': stocks[k]['total'] / portValue})
    print(portValue)

get_portfolio_value(stockData, 'total')

Answer 3

您的問題可以分兩部分解決：

1. 取總庫存的總和。 為此，您可以將sum()用作：

    stock_sum = sum(stock_data['total'] for stock_data in stockData.values())

2. 迭代該值以更新條目。 由於您不需要key ，請使用dict.values()僅迭代值：

    for stock_data in stockData.values(): stock_data['percentage'] = stock_data['total']/stock_sum # ^ Adds new key into your existing `dict` object

現在，您的stockData dict 的值以percentage持有附加鍵