如何根据指定值在DataFrame中生成有序行列序号列？

要创建包含row和column两列的DataFrame，且行数为指定值的平方，用Pandas可以快速实现，核心是生成重复的行号和循环的列号：

实现逻辑 #

• 设定指定值（比如N = 128）
• row列：每个数字从0到N-1依次重复N次
• column列：0到N-1的序列循环N次
• 将两列组合为DataFrame

代码示例 #

import pandas as pd

def create_grid_df(N):
    # 生成row列：每个行号重复N次
    rows = [i for i in range(N) for _ in range(N)]
    # 生成column列：列号循环N次
    cols = list(range(N)) * N
    # 构建DataFrame
    df = pd.DataFrame({'row': rows, 'column': cols})
    return df

# 测试：指定值为128
df_128 = create_grid_df(128)
# 验证行数（应为128*128=16384）
print(df_128.shape[0])

补充说明 #

• 若指定值为256，直接调用create_grid_df(256)即可，生成的DataFrame行数为256*256=65536
• row列呈现的是：0连续出现128次，接着1连续出现128次……直到127连续出现128次
• column列呈现的是：0到127的序列完整重复128轮

内容的提问来源于stack exchange，提问作者Miko