Data Science - Average of Rows

Question

Data Science - Average of Rows
Question:
In a matrix, or 2-d array X, the averages (or means) of the elements of rows is called row means.
Task
Given a 2D array, return the rowmeans.
Input Format
First line: two integers separated by spaces, the first indicates the rows of matrix X (n) and the second indicates the columns of X (p)
Next n lines: values of the row in X
Output Format
An numpy 1d array of values rounded to the second decimal.
2 2
1.5 1
2 2.9
Sample Output
[1.25 2.45]
Solution:
import numpy as np
n, p = [int(x) for x in input().split()]
lista = []
for i in range(n):
lista.append(input().split())
print(np.array(lista).astype(np.float16).mean(axis=1).round(2))
Solution fails test case:
Your output:
[1.5 0.75 0.65]
Expected output:
[1.5 0.75 0.65]
Notice the space before 0.75 means the test case fails. How may I remove the whitespace thanks.

Answer 1

please provide your test input to test

Answer 2

n, p = [int(x) for x in input().split()]

Answer 3

D33pesh
i said which input gave you the output
[1.5 0.75 0.65]

Answer 4

Input
3 2
1 2
1 0.5
1 0.3