A Python code which takes 2 numbers x and y as input and generates a 2-dimensional numpy array where value in the i-th row and j

Question

A Python code which takes 2 numbers x and y as input and generates a 2-dimensional numpy array where value in the i-th row and j-th column of the array should be (i+j)/2.
Note: i=0,1,...x-1 and j=0,1....,y-1
The input will have two lines with x and y respectively.
The output should be a 2D numpy array.
Sample Input:
3
4
Sample Output:

[[0. 0.5 1. 1.5]
[0.5 1. 1.5 2. ]
[1. 1.5 2. 2.5]]

Answer 1

import numpy as np
x=int(input())
y=int(input())
#write your code here
def d_array(r,c):
array = [[(i+j)/2 for j in range(y)] for i in range(x)]
a = np.matrix(array)
return a

print(d_array(x,y))

The Code worked for me

Answer 2

So what's your question?

Answer 3

Can you please simplify my code further ? I have a feel that it's more complex

Answer 4

Your code is as simple as it can be. What you could do is put it all in one line.

import numpy as np
x=int(input())
y=int(input())

def d_array(r,c):
return np.matrix([[(i+j)/2 for j in range(y)] for i in range(x)])

print(d_array(x,y))

Answer 5

x=int(input())
y=int(input())
arr = [[(i+j)/2 for j in range(y)] for i in range(x)]
print(np.matrix(arr))

Answer 6

Please show us your attempt.