我是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(性别)的函数?
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()
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句