如何以特殊方式將數據幀轉換為 Python 字典？

Question

df

      experiment    score
    generation_0.chrom0 11
    generation_0.chrom1 12
    generation_0.chrom2 13
    generation_0.chrom3 14

代碼：

trans_df = df.set_index("experiment").T
trans_df
a_dict = trans_df.to_dict("list")
print ( "Dictionary is ")
print ( a_dict)

Output：

字典是 {'generation_0.chrom0': [11], 'generation_0.chrom1': [12], 'generation_0.chrom2': [13], 'generation_0.chrom3': [14]}

我正在尋找在現有字典轉換中的兩個修改。

a）generation_0.chrom0 應該只是變成 chrom0 和其他人一樣

b）當前作為列表的值應該只是數字而不是列表

預期 Output ：

{'chrom0': 11, 'chrom1': 12, 'chrom2': 13, 'chrom3': 14}

Answer 1

dict(zip(df['experiment'].str.split('.').str[-1], df['score']))

Output

{'chrom0': 11, 'chrom1': 12, 'chrom2': 13, 'chrom3': 14}

Answer 2

跑：

result = pd.Series(df.score.values, index=df.experiment.str.split('.').str[-1]).to_dict()