我知道,如果我使用randn
,
import pandas as pd
import numpy as np
df = pd.DataFrame(np.random.randn(100, 4), columns=list('ABCD'))
给了我正在寻找的东西,但是有正态分布的元素。但是,如果我只想要随机整数怎么办?
randint
通过提供范围来工作,但不是像randn
那样的数组。那么如何在某个范围之间使用随机整数呢?
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
.. .. .. .. ..