从CSV文件中查找多列的中位数

Question

我需要创建一个函数，该函数将从csv数据集中获取每一列并返回其中值。例如：

def median_finder(column_names)

其中column_names是列的名称，输出是列的中位数。

我尝试过：

iris_data = pd.read_csv('iris.csv)
median = iris_data.median()
return median

median_finder('column name')

然后它给出了所有列的中位数，而不是我要求的一列：

sepal length (in cm)    5.80
sepal width (in cm)     3.00
petal length (in cm)    4.35
petal width (in cm)     1.30

我需要执行此功能，以便当我输入特定列时，它将返回该列的中位数，而不是所有列。谢谢您的帮助。

Answer 1

只需在数据框中选择一列即可

iris_data['nameofcolun'].median()

Answer 2

def iris_median(col):
    print(iris_data[col].median())