我使用pandas和numpy库来计算两个简单列表的pearson关联。以下代码的输出是相关矩阵:
import numpy as np
import pandas as pd
x = np.array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])
y = np.array([2, 1, 4, 5, 8, 12, 18, 25, 96, 48])
x, y = pd.Series(x), pd.Series(y)
xy = pd.DataFrame({'dist-values': x, 'uptime-values': y})
matrix = xy.corr(method="pearson")在对输出使用.unstack()和.to_dict()函数之后,我们可以使用以下格式的字典:
result = matrix.unstack().to_dict()
# {('dist-values', 'dist-values'): 1.0,
# ('dist-values', 'uptime-values'): 0.7586402890911869,
# ('uptime-values', 'dist-values'): 0.7586402890911869,
# ('uptime-values', 'uptime-values'): 1.0}但我需要将其转换为字典列表,输出应该如下所示:
#[ {'f1': 'dist-values', 'f2': 'dist-values', 'value': '1.0'},
# {'f1': 'dist-values', 'f2': 'uptime-values', 'value': '0.7586402890911869'},
# {'f1': 'uptime-values', 'f2': 'dist-values', 'value': '0.7586402890911869'},
# {'f1': 'uptime-values', 'f2': 'uptime-values', 'value': '1.0'}
# ]做这件事最好和有效的方法是什么?
发布于 2022-07-29 09:41:53
那麽:
result = (matrix.unstack().rename_axis(['f1', 'f2'])
.reset_index(name='value').to_dict('records')
)产出:
[{'f1': 'dist-values', 'f2': 'dist-values', 'value': 1.0},
{'f1': 'dist-values', 'f2': 'uptime-values', 'value': 0.7586402890911869},
{'f1': 'uptime-values', 'f2': 'dist-values', 'value': 0.7586402890911869},
{'f1': 'uptime-values', 'f2': 'uptime-values', 'value': 1.0}]https://stackoverflow.com/questions/73164541
复制相似问题