Python 3管道输入矩阵：兼容单行元素/整行输入的实现咨询

嘿，这个问题我太懂了！C语言里scanf()自动跳过空格、换行这些空白字符的特性确实省心，但Python虽然没有直接对应的内置函数，我们完全可以自己实现一样的效果，而且方法还挺灵活的。

核心思路：把所有输入统一处理成扁平化的数据流 #

不管输入是每行一个元素，还是每行一整行矩阵元素，我们的目标都是把所有输入的数字提取出来，按顺序排列——就像scanf()做的那样。下面两种方法都能实现这个需求：

方法一：一次性读取全部输入（简单直接） #

如果你的矩阵不是特别大，直接把所有输入读进来再分割是最省事的：

import sys

# 读取所有输入内容，按任意空白字符（空格、换行、制表符）分割成字符串列表，再转成整数
all_input = list(map(int, sys.stdin.read().split()))

# 提取列数、行数和矩阵元素
cols = all_input[0]
rows = all_input[1]
matrix_values = all_input[2:]

# 把一维列表转换成二维矩阵
matrix = [matrix_values[i*cols : (i+1)*cols] for i in range(rows)]

# 测试输出
print(f"列数: {cols}, 行数: {rows}")
print("矩阵内容：")
for row in matrix:
    print(row)

这段代码里，sys.stdin.read()会把管道或文件输入的所有内容一次性读成一个字符串，split()默认会分割所有空白字符——不管是空格还是换行，完全模拟了scanf()的行为。

方法二：逐行读取（适合超大矩阵，节省内存） #

如果你的矩阵特别大，一次性读全部内容可能会占用太多内存，那就逐行处理，每次把当前行的元素加到总列表里：

import sys

all_input = []
for line in sys.stdin:
    # 分割当前行的元素，转成整数后添加到总列表
    all_input.extend(map(int, line.split()))

# 后续处理和方法一完全一致
cols = all_input[0]
rows = all_input[1]

# 校验元素数量是否匹配，避免格式错误
if len(all_input) != 2 + cols * rows:
    print("错误：输入的元素数量与行列数不匹配！")
    sys.exit(1)

matrix = [all_input[2 + i*cols : 2 + (i+1)*cols] for i in range(rows)]
print(matrix)

这种方式每次只处理一行数据，内存占用小很多，同时也能兼容两种输入格式。

额外提示 #

• 记得加格式校验：比如检查元素总数是否等于cols * rows，避免因为输入格式错误导致程序崩溃。
• 如果输入里可能混有非数字内容，可以用try-except块捕获ValueError，给用户更友好的错误提示：

try:
    all_input = list(map(int, sys.stdin.read().split()))
except ValueError:
    print("错误：输入包含非数字内容！")
    sys.exit(1)

内容的提问来源于stack exchange，提问作者7_R3X