如何从DataFrame中随机选择一位用户或不选择任何用户？

现有如下pandas DataFrame：

import pandas as pd
df=pd.DataFrame({'user': [1, 2, 3, 4, 5, 6],
                'value': [0.64, 0.93, 0, 0.43, 0.65, 0]})

目前已使用sample()方法随机选取一行（一位用户）：

df1 = df.sample()

现需实现：随机选择一位用户（如1、2、3、4、5、6中的某一个）并输出user X is selected，或不选择任何用户并输出no user is selected，该如何操作？

核心是先随机决定「是否选择用户」，再执行对应操作，下面提供两种实用实现方式：

方法一：固定概率二选一判断 #

用random.choice()直接敲定是选用户还是不选，逻辑简单直观：

import pandas as pd
import random

df=pd.DataFrame({'user': [1, 2, 3, 4, 5, 6],
                'value': [0.64, 0.93, 0, 0.43, 0.65, 0]})

# 50%概率选用户，50%概率不选
decision = random.choice(['select', 'no_select'])

if decision == 'select':
    selected_user = df.sample()['user'].iloc[0]
    print(f"user {selected_user} is selected")
else:
    print("no user is selected")

方法二：自定义选择概率 #

可以灵活设置选择用户的概率（比如70%概率选，30%不选）：

import pandas as pd
import random

df=pd.DataFrame({'user': [1, 2, 3, 4, 5, 6],
                'value': [0.64, 0.93, 0, 0.43, 0.65, 0]})

# 设置选择用户的概率，0.7代表70%概率触发选择
select_prob = 0.7

if random.random() < select_prob:
    selected_user = df.sample()['user'].iloc[0]
    print(f"user {selected_user} is selected")
else:
    print("no user is selected")

补充：排除value为0的用户 #

如果需要跳过value为0的用户（比如用户3、6），可以先过滤再采样：

import pandas as pd
import random

df=pd.DataFrame({'user': [1, 2, 3, 4, 5, 6],
                'value': [0.64, 0.93, 0, 0.43, 0.65, 0]})

select_prob = 0.7
# 过滤掉value为0的用户
filtered_df = df[df['value'] != 0]

if filtered_df.empty:
    print("no available user to select")
elif random.random() < select_prob:
    selected_user = filtered_df.sample()['user'].iloc[0]
    print(f"user {selected_user} is selected")
else:
    print("no user is selected")

内容的提问来源于stack exchange，提问作者alxander21