gpt4 book ai didi

python - 融化 Pandas 数据框的上三角矩阵

转载 作者:IT老高 更新时间:2023-10-28 20:33:00 35 4
gpt4 key购买 nike

给定一个如下形式的方形 pandas DataFrame:

   a  b  c
a 1 .5 .3
b .5 1 .4
c .3 .4 1

upper triangle 怎么可能?被熔化得到如下形式的矩阵

 Row     Column    Value
a a 1
a b .5
a c .3
b b 1
b c .4
c c 1

#Note the combination a,b is only listed once. There is no b,a listing

我对惯用的 pandas 解决方案更感兴趣,自定义索引器很容易手动编写...

提前感谢您的考虑和回复。

最佳答案

首先,我通过 wheredf 的较低值转换为 NaNnumpy.triu然后 stack , reset_index并设置列名:

import numpy as np

print df
a b c
a 1.0 0.5 0.3
b 0.5 1.0 0.4
c 0.3 0.4 1.0

print np.triu(np.ones(df.shape)).astype(np.bool)
[[ True True True]
[False True True]
[False False True]]

df = df.where(np.triu(np.ones(df.shape)).astype(np.bool))
print df
a b c
a 1 0.5 0.3
b NaN 1.0 0.4
c NaN NaN 1.0

df = df.stack().reset_index()
df.columns = ['Row','Column','Value']
print df

Row Column Value
0 a a 1.0
1 a b 0.5
2 a c 0.3
3 b b 1.0
4 b c 0.4
5 c c 1.0

关于python - 融化 Pandas 数据框的上三角矩阵,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/34417685/

35 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com