random.randint()

126 阅读1分钟

使用 NumPy 的 random.randint() 函数来生成指定范围内的随机整数,并将其添加为 DataFrame 的新列。以下是如何实现的示例代码:

import numpy as np
import pandas as pd

# 假设 data 是你的 DataFrame,现在添加一个名为 "count" 的随机数列
data["count"] = np.random.randint(1, 6, size=len(data))

# 如果你不想覆盖原始的 DataFrame,可以通过以下方式创建一个副本并添加新列
data_with_random = data.copy()
data_with_random["count"] = np.random.randint(1, 6, size=len(data))

# 显示添加了随机数列的 DataFrame
print(data_with_random)

这样就可以在 DataFrame 中添加一个名为 "count" 的列,并填充随机整数值(范围在 1 到 5 之间)。

np.random.randint() 函数的 size 参数用于指定生成随机数的数量,它的默认值是 None。当 size=None 时,函数只返回一个随机数;当 size 是一个整数时,函数返回一个由指定大小的随机数数组组成的数组。如果你想要生成的随机数数量与 DataFrame 的行数相同,你可以将 size 设置为 DataFrame 的行数,这样可以确保每行都有一个对应的随机数。

在这里,假设 data 是你的 DataFrame,len(data) 表示 DataFrame 的行数,因此将 size 设置为 len(data) 可以确保生成的随机数数量与 DataFrame 的行数相同,从而为每一行添加一个随机数。