Как создать DataFrame случайных целых чисел с Pandas?

Я знаю, что если я использую randn,

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

дает мне то, что я ищу, но с элементами из нормального распределения. Но что, если я просто хочу случайные целые числа?

randint работает, предоставляя диапазон, но не массив, как randn. Так как мне сделать это со случайными целыми числами между некоторым диапазоном?

Ответы

Ответ 1

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
..  ..  ..  ..  ..