按标准拆分列表(性别)

2024-03-29 05:27:40 发布

您现在位置:Python中文网/ 问答频道 /正文

我是python的初学者,只是在学习如何编写函数。我已经得到了一个权重和性别的列表,并试图根据性别标准将其拆分以创建两个新的列表。使用for循环,到目前为止我已经取得了成功:

df = pd.read_csv('brainweight.csv')
w = list(df['Weight'])
s = list(df['Sex'])

female_weight = []
male_weight = []

for sex, weight in zip (s, w):
    if sex == 'f':
        female_weight.append(weight)
    else:
        male_weight.append(weight)

我应该如何将这个for循环修改为变量=m/f(gender)的函数?你知道吗


Tags: csv函数df列表for标准malefemale
1条回答
网友
1楼 · 发布于 2024-03-29 05:27:40
df = pd.read_csv('brainweight.csv')
w = list(df['Weight'])
s = list(df['Sex'])
def list_by_gender():
    females = []
    males = []
    for sex, weight in zip (s, w):
        if sex == 'f':
            females.append(weight)
        else:
            males.append(weight)
    return males,females

male_list,female_list=list_by_gender()

相关问题 更多 >