要实现不用逐个输入列名进行计算多列均值的功能,可以使用变长参数(*args)来接收多个列名,然后通过变异(apply)函数来计算均值。
下面是一个示例代码:
import pandas as pd
def calculate_mean(*args):
# 读取数据集
df = pd.read_csv('data.csv')
# 使用变异函数计算多列的均值
mean_values = df[args].apply(lambda x: x.mean(), axis=0)
return mean_values
# 调用函数并传入多个列名
result = calculate_mean('列名1', '列名2', '列名3')
print(result)
在上面的代码中,calculate_mean
函数使用了变长参数(*args)来接收多个列名。然后,通过pd.read_csv
函数读取数据集,并使用df[args]
来选择指定的多列数据。最后,使用apply
函数和lambda
表达式来计算每列的均值。
你可以根据实际情况修改示例代码中的数据集读取方式和输出方式。