如何使用Pandas创建随机整数的DataFrame?

问题描述 投票:72回答:1

我知道,如果我使用randn

import pandas as pd
import numpy as np
df = pd.DataFrame(np.random.randn(100, 4), columns=list('ABCD'))

给了我正在寻找的东西,但是有正态分布的元素。但是,如果我只想要随机整数怎么办?

randint通过提供范围来工作,但不是像randn那样的数组。那么如何在某个范围之间使用随机整数呢?

python pandas dataframe size shapes
1个回答
116
投票

numpy.random.randint接受第三个参数(size),您可以在其中指定输出数组的大小。你可以用它来创建你的DataFrame -

df = pd.DataFrame(np.random.randint(0,100,size=(100, 4)), columns=list('ABCD'))

这里 - np.random.randint(0,100,size=(100, 4)) - 在(100,4)之间创建一个大小为[0,100)的输出数组,其中包含随机整数元素。


演示 -

import numpy as np
import pandas as pd
df = pd.DataFrame(np.random.randint(0,100,size=(100, 4)), columns=list('ABCD'))

产生:

     A   B   C   D
0   45  88  44  92
1   62  34   2  86
2   85  65  11  31
3   74  43  42  56
4   90  38  34  93
5    0  94  45  10
6   58  23  23  60
..  ..  ..  ..  ..
© www.soinside.com 2019 - 2024. All rights reserved.