希望在python3+中只使用一个用户定义的输入和输出函数

Question

from numpy import*
def row():
for i in range(len(a)):
for j in range(len(a[i])):
t = [i, j]
inp(*t)
def inp(*m):
a[m] = int(input(f"entert the element of {m} = "))
out(*m)
def out(*o):
print(a[o])
a = zeros((1,2), dtype = int)
row()

输出显示如下：

enter the element of (0, 0) = 2
2
enter the element of (0, 1) = 3
3

但我想显示像这个一样的输出

在类似的时间输入

enter the element of (0, 0) = 2
enter the element of (0, 1) = 3

在类似的时间输出

2
3

可以通过创建两个单独的for循环或两个用于输入和输出的函数来实现但我的问题是：我怎么能只创建一个功能

def row():
for i in range(len(a)):
for j in range(len(a[i])):
t = [i, j]
inp(*t)

Answer 1

不要立即打印输出，而是保存值并在收集完后打印：

import numpy as np
def row(a):
values = []
for i in range(len(a)):
for j in range(len(a[i])):
value = get_input(a, i, j)
values.append(value)
for value in values:
print(value)
def get_input(a, *m):
a[m] = int(input(f"enter the element of {m} = "))
return a[m]
a = np.zeros((1,2), dtype = int)
row(a)

给出

enter the element of (0, 0) = 1
enter the element of (0, 1) = 2
1
2

然而，还有一种更简单的方法：因为你也可以在矩阵中设置新元素，所以你不必单独保存它们，只需打印出矩阵值：

for item in a.ravel():
print(item)