熊猫 - 获取与条目相关的密钥

Question

我有一个平方相关矩阵：

a0 a1 a2 ……… an
b1
b2
……
bn

我想将这个矩阵的多索引中的键传递给另一个函数，所以想要形成所有密钥对的元组列表（通过过滤函数，比如说（e＆gt; = .95）< / p>

[('KEYA', 'KEYB')]

是否有一个内置于熊猫的功能可以让我实现这个目标？ RTMF已经让我失望了几个小时。

Answer 1

如果要创建过滤，则可以先堆叠数据帧，然后将其过滤到所需的值，然后将索引强制为列表。

df1 = df.stack()
df_final = df1[df1 > .95]
df_final.index.tolist()

创建虚假数据和测试

np.random.seed(12)
df = pd.DataFrame(np.random.rand(10,10))

df1 = df.stack()
df_final = df1[df1 > .95]
df_final.index.tolist()

会导致

[(0, 8), (3, 3), (3, 5), (4, 1), (4, 3), (9, 0)]

df_final看起来像这样：

0  8    0.956949
3  3    0.978058
   5    0.950314
4  1    0.995138
   3    0.962597
9  0    0.957417