如何从嵌套循环中生成的数据创建热图/ Pandas数据框架?

Question

嘿，我想从嵌套循环中生成的数据创建热图(或Pandas Dataframe)，如下例所示:

for i in range (1,5):
a = i + 2
for n in range(1,5):
c = n - 1

所以我想要的输出是在一个轴上有i，在另一个轴上有b作为两者的值相交-比如a+c:

i 
1   2   3   4   5
1   (3) (4)
2   (4)
n   3   (5) 
4
5

括号中的数为a+c的结果。我认为我应该创建一个列表，并将这些计算的结果附加在它们后面，但我不知道具体怎么做。任何建议都会有帮助的!

---

p。这是简化的情况，在我的情况下，这必须是嵌套循环-我想检查两个特性的变化如何影响预测

Answer 1

我发布这个是为了展示代码示例。我不认为这是对你问题的完整答案，但这是正确的方向。

import pandas as pd
import numpy as np
data = np.zeros((5,5), dtype=np.int32)
for i in range(5):
for n in range(5):
data[n,i] = (i+1+2) + (n+1-1)
print(data)
df = pd.DataFrame(data)
print(df)

Answer 2

我有点难以理解你在哪里使用c，但这似乎正在做你正在寻找的:

df = pd.DataFrame(columns=[1,2,3,4,5])
for i in range(1,6):
df.loc[len(df.index)] = list(a+(i+1))
df.index += 1

输出:

1   2   3   4   5
1   3   4   5   6   7
2   4   5   6   7   8
3   5   6   7   8   9
4   6   7   8   9   10
5   7   8   9   10  11