Computer >> Máy Tính >  >> Lập trình >> Python

Viết mã Python để chọn bất kỳ một hàng ngẫu nhiên nào từ DataFrame nhất định

Đầu vào -

Giả sử, DataFrame mẫu là,

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

Đầu ra -

Random row is
  Id    5
Name Peter

Giải pháp

Để giải quyết vấn đề này, chúng tôi sẽ làm theo các cách tiếp cận dưới đây.

  • Xác định DataFrame

  • Tính số hàng bằng cách sử dụng df.shape [0] và gán cho biến hàng.

  • đặt giá trị random_row từ phương thức randrange như được hiển thị bên dưới.

random_row = r.randrange(rows)
  • Áp dụng random_row bên trong cắt lát iloc để tạo bất kỳ hàng ngẫu nhiên nào trong DataFrame. Nó được định nghĩa bên dưới,

df.iloc[random_row,:]

Ví dụ

Hãy cùng chúng tôi xem cách triển khai sau để hiểu rõ hơn.

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,:])

Đầu ra

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