How can I solve the Machine learning "The Basics" project?

Question

I have this code right here

import pandas as pd
df = pd.read_csv('/usercode/files/one.csv')
print(df['a'].values)
and it works well for the first test case but I can't seem to find a workaround the other test cases.
Much appreciated

Answer 1

Here is the code I used to solve the challenge. I hope it helps :)

import pandas as pd

filename = input()
column_name = input()

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

Answer 2

You are welcome 🙂

Answer 3

WOW, I can't believe how easy you made it look, thank you so much!!!