我正在嘗試使用 DataFrame 中的索引作為鍵來計算字典中特定值的實體數。然而，字典鍵集中缺少一些 DataFrame 索引。如何構建字典理解來克服這個問題？
import pandas as pd
df = pd.DataFrame({'Date': ['2022-02-14', '2022-02-14', '2022-02-14', '2022-02-14', '2022-02-14'],
'count': [10, 10, 10, 9, 9],},
index = ['NNI', 'NVEC', 'IPA', 'LYTS', 'MYN'])
df
Date count
NNI 2022-02-14 10
NVEC 2022-02-14 10
IPA 2022-02-14 10
LYTS 2022-02-14 9
MYN 2022-02-14 9
dct = {'NNI' : pd.DataFrame({'s': [-1, -1, -1],
'count': [13, 11, 10]},
index =['2007-07-13', '2019-09-18', '2016-08-01']),
'NVEC' : pd.DataFrame({'s': [-1, -1, -1],
'count': [12, 10, 9]},
index =['2012-10-09', '2018-10-01', '2022-02-01'])
}
dct
{'NNI': s count
2007-07-13 -1 13
2019-09-18 -1 11
2016-08-01 -1 10,
'NVEC': s count
2012-10-09 -1 12
2018-10-01 -1 10
2022-02-01 -1 9}
我沒有運氣就嘗試了以下方法。
{k:len(dct[k]['count']) for k in df.index if k in df.index}
KeyError Traceback (most recent call last)
<ipython-input-201-10244dbbde59> in <module>
----> 1 {k:len(dct[k]['count']) for k in df.index if k in df.index}
<ipython-input-201-10244dbbde59> in <dictcomp>(.0)
----> 1 {k:len(dct[k]['count']) for k in df.index if k in df.index}
KeyError: 'IPA'
uj5u.com熱心網友回復：
如果我理解正確，您的if條款應該涉及dct而不是df.index
{k:len(dct[k]['count']) for k in df.index if k in dct.keys()}
應該管用。
如果您的 dict 非常大 - 您可能希望先獲取所有鍵，然后對其進行迭代，例如：
dct_keys = dct.keys()
{k:len(dct[k]['count']) for k in df.index if k in dct_keys}
轉載請註明出處，本文鏈接：https://www.uj5u.com/ruanti/424749.html