关于列数据等的机器学习 Sololearn Coach 问题

Question

问题：

机器学习 - 列中有什么？

从 numpy 数组中获取一列。

任务 - 给定一个 csv 文件和一个列名，打印给定列中的元素。

输入格式 - 第一行：csv 文件的文件名； 第二行：列名

输出格式 - numpy 数组

示例输入 - 'usercode/files/one.csv' (文件名) ; 'a' (列名)

文件“one.csv”内容：

a,b

1,3

2,4

样本输出 - [1 2]

---------

我的回答：

import pandas as p 
df = p.read_csv('usercode/files/one.csv')

details = df[['a', 'b']].values

print(details[:,1])

但是，

我认为它需要输出 [1,2] 和 [3,4] 以某种方式满足 Case 1 和案例2同时进行。我的代码不能这样做。如果我满足 Case 1，则 Case 2 不满足，反之亦然

Answer 1

import pandas as pd


filename = input()
column_name = input()

df = pd.read_csv(filename)
arr = df[column_name].values

print(arr)

filename 和 column_name 是用户将输入的参数，只需将这些参数写入函数中即可得到答案。