[已解决]运算结果保持在矩阵的一个三角形内，应如何办？请大师帮忙，谢谢

futui

下面是我的期望，运算结果保持在矩阵的一个三角形内，应如何修改我的代码？请大师帮忙，谢谢


index    11         22          33
0    11     1        1122       1133
1    22                 1           2233
2    33                              1


import pandas as pd

shtcode = ['11','22','33']
df = pd.DataFrame(0, index=shtcode, columns=shtcode)
df.reset_index(inplace=True)

for index, row in df.iterrows():
    a = row[df.columns[0]]
    for b in df.columns[1:]:
        if a == b:
            df.loc[index, b] = 1
        else:
            df.loc[index, b] = a + b


print(df)

最佳答案

月排行榜 / 总排行榜

歌者文明清理员

2023-5-3 17:33:36

您可以在循环时加一个判断来保持结果在矩阵的一个三角形内。具体方法是：对于每个行号大于等于列号的元素，进行运算并赋值，否则直接跳过。以下是修改后的代码：

复制代码

1. import pandas as pd
   
2. 
   
3. shtcode = ['11','22','33']
   
4. df = pd.DataFrame(0, index=shtcode, columns=shtcode)
   
5. 
   
6. for i in range(len(shtcode)):
   
7.     for j in range(i, len(shtcode)):
   
8.         a = shtcode[i]
   
9.         b = shtcode[j]
   
10.         if a == b:
    
11.             df.loc[a, b] = 1
    
12.         else:
    
13.             df.loc[a, b] = a + b
    
14. 
    
15. print(df)

复制代码

输出结果如下，满足要求：

11    22    33
11  1   1122  1133
22       1    2233
33            1

跳转到最佳答案楼层

isdkz

你可以使用以下代码来保持矩阵的一个三角形：

1. import pandas as pd
   
2. 
   
3. shtcode = ['11','22','33']
   
4. df = pd.DataFrame(0, index=shtcode, columns=shtcode)
   
5. 
   
6. for i in range(len(shtcode)):
   
7.     for j in range(i, len(shtcode)):
   
8.         if i == j:
   
9.             df.iloc[i,j] = 1
   
10.         else:
    
11.             df.iloc[i,j] = shtcode[i] + shtcode[j]
    
12. 
    
13. # 对角线下三角形复制到对角线上三角形
    
14. for i in range(len(shtcode)):
    
15.     for j in range(i):
    
16.         df.iloc[j,i] = df.iloc[i,j]
    
17. 
    
18. print(df)
    

复制代码

这里使用了两个嵌套循环来遍历矩阵的每个元素。当 i=j 时，将对角线上的元素设置为1。否则，将元素设置为相应的字符串连接。最后，将对角线下三角形中的元素复制到对角线上三角形中。

希望这可以帮助你实现你的期望。

歌者文明清理员

您可以在循环时加一个判断来保持结果在矩阵的一个三角形内。具体方法是：对于每个行号大于等于列号的元素，进行运算并赋值，否则直接跳过。以下是修改后的代码：

复制代码

1. import pandas as pd
   
2. 
   
3. shtcode = ['11','22','33']
   
4. df = pd.DataFrame(0, index=shtcode, columns=shtcode)
   
5. 
   
6. for i in range(len(shtcode)):
   
7.     for j in range(i, len(shtcode)):
   
8.         a = shtcode[i]
   
9.         b = shtcode[j]
   
10.         if a == b:
    
11.             df.loc[a, b] = 1
    
12.         else:
    
13.             df.loc[a, b] = a + b
    
14. 
    
15. print(df)

复制代码

输出结果如下，满足要求：

11    22    33
11  1   1122  1133
22       1    2233
33            1

sfqxx

根据您的期望，您可以修改代码以确保矩阵中的运算结果仅保留在三角形内。此外，还需要考虑对角线上的元素应该为1。以下是修改后的代码：

1. 
   
2. import pandas as pd
   
3. 
   
4. shtcode = ['11','22','33']
   
5. df = pd.DataFrame(0, index=shtcode, columns=shtcode)
   
6. df.reset_index(inplace=True)
   
7. 
   
8. for i in range(len(df)):
   
9.     for j in range(i+1, len(df)):
   
10.         df.iloc[i,j] = df.iloc[i,j-1] + df.iloc[j-1,j]
    
11.         
    
12. # Set diagonal elements to 1
    
13. df.loc[df.index, df.columns] = 1
    
14. print(df)

复制代码

主要修改包括：

1. 替换了df.iterrows()，使用双层循环的方式来遍历数据框中的元素。这样可以更方便地控制迭代的范围。
2. 内循环仅处理三角形中的元素，并通过调整索引值来访问每个元素。
3. 将对角线元素设置为1。
有用请设置最佳答案

sfqxx

歌者文明清理员 发表于 2023-5-3 17:33
您可以在循环时加一个判断来保持结果在矩阵的一个三角形内。具体方法是：对于每个行号大于等于列号的元素， ...

你好幸运啊

歌者文明清理员

sfqxx 发表于 2023-5-3 17:36
根据您的期望，您可以修改代码以确保矩阵中的运算结果仅保留在三角形内。此外，还需要考虑对角线上的元素应 ...

sfqxx

歌者文明清理员 发表于 2023-5-3 17:47

突然我也有你之前的感受了

歌者文明清理员

sfqxx 发表于 2023-5-3 17:47
突然我也有你之前的感受了