如何以特殊方式将数据帧转换为 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()