Python 您可以根据以下条件添加行

我有一个整数

n=3

我必须创建一个正方形矩阵 当矩阵[行]=矩阵[列]时，使用以下条件

，然后使用1或0

我的计划是创建一个表，首先插入所有空值的

n

行数，然后用上述case语句在sql中编写一个更新

在这种情况下，任何人都可以帮助创建一个最初带有

n

行和

n

列的数据帧吗

---

谢谢。

您可以使用

numpy

和

pandas

执行类似操作：

import numpy as np
import pandas as pd

def create_diagonal_matrix_of_size(n):
    if n > 0:
        d = np.diag(n * [1])
        df = pd.DataFrame(d)
        return df
    else:
        return pd.DataFrame()

print(create_diagonal_matrix_of_size(3))

---

谢谢，如果可能的话，我只想使用Pandas和SQL概念。谢谢你的回答和快速响应。Pandas使用numpy。所以，如果你正在使用熊猫，你已经可以访问numpy了。很抱歉，如果我在下面这样做，我会问一个愚蠢的问题：因为我在（0，n，1）范围内。。。a、 为范围（0，len（a），1）中的i追加（n）：c.append（[a]）我可以在上面创建一个数据帧并编写一个sql更新吗。。。。使用Spark sql可能。。。我只是时间复杂度和空间的古玩。。。。。