编写 Python 代码从给定的 DataFrame 中选择任意一行随机数据

pythonpandasserver side programmingprogramming

输入

假设,示例 DataFrame 为,

 Id Name
0 1 Adam
1 2 Michael
2 3 David
3 4 Jack
4 5 Peter

输出output

随机行为
  Id    5
姓名 Peter

解决方案

为了解决这个问题,我们将遵循以下方法。

  • 定义 DataFrame

  • 使用 df.shape[0] 计算行数并分配给 rows 变量。

  • 从 randrange 方法设置 random_row 值,如下所示。

random_row = r.randrange(rows)
  • 在 iloc 切片中应用 random_row 以生成 DataFrame 中的任意随机行。它的定义如下,

df.iloc[random_row,:]

示例

让我们看看下面的实现,以便更好地理解。

import pandas as pd
import random as r
data = { 'Id': [1,2,3,4,5],'Name': ['Adam','Michael','David','Jack','Peter']}
df = pd.DataFrame(data)
print("DataFrame is\n", df)
rows = df.shape[0]
print("total number of rows:-", rows)
random_row = r.randrange(rows)
print("print any random row is\n")
print(df.iloc[random_row,:])

输出

DataFrame is
 Id Name
0 1 Adam
1 2 Michael
2 3 David
3 4 Jack
4 5 Peter
total number of rows:- 5
print any random row is
  Id   3
Name David

相关文章