📜  在 Pandas Dataframe 中生成随机整数(1)

📅  最后修改于: 2023-12-03 15:37:25.473000             🧑  作者: Mango

在 Pandas Dataframe 中生成随机整数

在 Pandas 中,我们可以使用 numpy 库提供的 random 模块生成随机整数,并将其存储在 Dataframe 中。具体实现方法如下:

步骤1:导入必要的库

我们首先需要导入 numpypandas 库。

import numpy as np
import pandas as pd
步骤2:生成随机整数

接下来,我们使用 numpy 库提供的 random.randint 方法生成随机整数。

# 生成 5 行 3 列的随机整数数组
random_data = np.random.randint(low=0, high=100, size=(5, 3))

该方法接受三个参数:

  • low:生成的随机整数的最小值。
  • high:生成的随机整数的最大值(不包含)。
  • size:生成的随机整数数组的大小。

在这个例子中,我们生成了一个大小为 5 × 3 的数组,其中元素的值在 0 到 100 之间取值。

步骤3:将随机整数存储在 Dataframe 中

最后,我们使用 Pandas 提供的 DataFrame 类将随机整数存储在 Dataframe 中。

# 将随机整数存储在 Dataframe 中
df = pd.DataFrame(random_data, columns=['A', 'B', 'C'])

在这个例子中,我们将随机整数存储在一个大小为 5 × 3 的 Dataframe 中,并将列的名称分别设置为 A、B、C。

完整代码
import numpy as np
import pandas as pd

# 生成 5 行 3 列的随机整数数组
random_data = np.random.randint(low=0, high=100, size=(5, 3))

# 将随机整数存储在 Dataframe 中
df = pd.DataFrame(random_data, columns=['A', 'B', 'C'])

print(df)

输出结果:

    A   B   C
0  16  70  85
1  95   6  49
2  90  77  30
3  59  83  51
4  26  50  32

通过以上步骤,我们就可以在 Pandas Dataframe 中生成随机整数了。