Sử dụng Pandas để lấy mẫu DataFrame sử dụng trọng lượng của một cột cụ thể của

Question

Tôi có một DataFrame mà trông giống như:

index name city 
    0  Yam Hadera 
    1  Meow Hadera 
    2  Don Hadera 
    3  Jazz Hadera 
    4  Bond Tel Aviv 
    5  James Tel Aviv 

Tôi muốn Pandas để lựa chọn một cách ngẫu nhiên các giá trị, bằng cách sử dụng số lần ra sân trong cột city (loại sử dụng: df.city.value_counts()), do đó kết quả của chức năng kỳ diệu của tôi, giả sử:

df.magic_sample(3, weight_column='city') 

có thể trông giống như:

0  Yam  Hadera 
    1  Meow  Hadera 
    2  Bond  Tel Aviv 

Cảm ơn! :)

Answer 1

Bạn có thể nhóm bởi city và sau đó lấy mẫu từng nhóm dựa trên độ dài của chúng so với chiều dài của khung dữ liệu gốc:

df.groupby('city', group_keys=False).apply(lambda g: g.sample(3 * len(g)/len(df))) 

Answer 2

Nếu tôi hiểu câu hỏi một cách chính xác, có lẽ bạn đang tìm kiếm random.sample:

>>> import pandas as pd 
>>> from random import sample 
>>> df = pd.DataFrame(data=[('Yam', 'Hadera'), ('Meow', 'Hadera'), ('Don', 'Hadera'), ('Jazz', 'Hadera'), ('Bond', 'Tel Aviv'), ('James', 'Tel Aviv')], columns=('name', 'city')) 
>>> df 
    name  city 
0 Yam Hadera 
1 Meow Hadera 
2 Don Hadera 
3 Jazz Hadera 
4 Bond Tel Aviv 
5 James Tel Aviv 
>>> df.iloc[sample(range(len(df)), 3), :] 
    name  city 
4 Bond Tel Aviv 
0 Yam Hadera 
1 Meow Hadera