python - 动态 pandas 数据帧生成

Question

这是我编写的用于生成包含 4 列的数据框的代码

num_rows = 10

df = pd.DataFrame({ 'id_col' : [x+1 for x in range(num_rows)] , 'c1': [randint(0, 9) for x in range(num_rows)], 'c2': [randint(0, 9) for x in range(num_rows)], 'c3': [randint(0, 9) for x in range(num_rows)] })
df

print(df) 渲染:

id_col  c1  c2  c3
0       1   3   1   5
1       2   0   2   4
2       3   1   2   5
3       4   0   5   6
4       5   0   0   1
5       6   6   5   8
6       7   1   6   8
7       8   5   8   8
8       9   1   5   2
9      10   2   9   2

我已设置通过 num_rows 变量动态生成的行数。

如何动态生成 1000 列，其中每列前面都带有“c”。因此生成列 c1,c2,c3....c1000，其中每列包含 10 行？

Answer 1

为了获得更好的性能，我建议使用创建 DataFrame numpy 函数 numpy.random.randint然后通过列表理解更改列名称，对于按位置的新列，请使用 DataFrame.insert :

np.random.seed(458)

N = 15
M = 10
df = pd.DataFrame(np.random.randint(10, size=(M, N)))
df.columns = ['c{}'.format(x+1) for x in df.columns]
df.insert(0, 'idcol', np.arange(M))

print (df)
   idcol  c1  c2  c3  c4  c5  c6  c7  c8  c9  c10  c11  c12  c13  c14  c15
0      0   8   2   1   6   2   1   0   9   7    8    0    5    5    6    0
1      1   0   2   5   0   0   2   5   2   9    2    1    0    0    5    0
2      2   5   1   3   5   4   5   3   0   2    1    7    8    9    5    4
3      3   8   7   7   0   1   3   6   7   5    8    8    9    8    5    5
4      4   2   8   1   7   3   7   4   6   0    7    0    9    4    0    4
5      5   9   2   1   6   1   9   5   6   7    4    6    1    7    3    7
6      6   1   9   3   9   7   7   2   7   9    8    2    7    2    5    5
7      7   7   6   6   6   4   2   9   0   6    5    7    0    0    4    9
8      8   6   4   2   1   3   1   7   0   4    3    0    5    4    7    7
9      9   1   3   5   7   2   2   1   5   6    1    9    5    9    6    3

另一个解决方案 numpy.hstack对于堆栈第一个 id 列到二维数组:

np.random.seed(458)

arr = np.hstack([np.arange(M)[:, None], np.random.randint(10, size=(M, N))])
df = pd.DataFrame(arr)
df.columns = ['idcol'] + ['c{}'.format(x) for x in df.columns[1:]]
print (df)
   idcol  c1  c2  c3  c4  c5  c6  c7  c8  c9  c10  c11  c12  c13  c14  c15
0      0   8   2   1   6   2   1   0   9   7    8    0    5    5    6    0
1      1   0   2   5   0   0   2   5   2   9    2    1    0    0    5    0
2      2   5   1   3   5   4   5   3   0   2    1    7    8    9    5    4
3      3   8   7   7   0   1   3   6   7   5    8    8    9    8    5    5
4      4   2   8   1   7   3   7   4   6   0    7    0    9    4    0    4
5      5   9   2   1   6   1   9   5   6   7    4    6    1    7    3    7
6      6   1   9   3   9   7   7   2   7   9    8    2    7    2    5    5
7      7   7   6   6   6   4   2   9   0   6    5    7    0    0    4    9
8      8   6   4   2   1   3   1   7   0   4    3    0    5    4    7    7
9      9   1   3   5   7   2   2   1   5   6    1    9    5    9    6    3