Solving Machine Learning Course Code Project

Question

Hey there, I am solving the first code project in the machine learning course and having a problem.
There are a total of 5 test cases. But the problem is that once I solve the first one It says that my output does not match the second test case.
How can I solve all 5 test cases at the same time?! I am confused here.
I first try to solve the first test case with this code:
"import pandas as pd
import numpy as np
df = pd.read_csv("/usercode/files/one.csv")
arr = np.array(df)
n1 = arr[0][0]
n2 = arr[1][0]
a = np.array([n1, n2])
print(a)"
and It works but suddenly It says that the output does not match the second test case and once I solve the second test case with similar code It says that the output does not match the desired output of the first test case.
How do I give an output that is exactly the same for all test cases at the same time?! How does this work?!
If I sound irritating please forgive me as I am a newbie.
Happy coding :)

Answer 1

Refer https://code.sololearn.com/cGzmSVPdv3kZ/?ref=app
Hope this helps.