Ввод массива с клавиатуры в Python: от базовых списков до сложных структур

Работа с массивами данных — основа программирования на Python, и умение корректно заполнять их через пользовательский ввод открывает двери для создания интерактивных программ. Однако многие новички сталкиваются с проблемами: как преобразовать строку из input() в числовой массив? Как избежать ошибок при вводе многомерных структур? И почему иногда программа падает с загадочной ошибкой ValueError?

Эта статья не просто покажет как ввести массив с клавиатуры в Python, но и раскроет нюансы обработки данных: от простейших списков чисел до вложенных структур с проверкой корректности. Мы разберём 5 практических методов (включая малоизвестные трюки с map() и генераторами), проанализируем типичные ошибки и оптимизируем код для реальных задач — от учебных примеров до парсинга пользовательского ввода в ботах.

Особое внимание уделим обработке некорректного ввода: как сделать программу устойчивой к опечаткам, лишним пробелам или несоответствию типов данных. Эти навыки критичны для создания надёжных скриптов, которые не «падают» при первом же неверном символе.

1. Базовый ввод одномерного массива: от строки к числам

Начнём с классической задачи: пользователь вводит числа через пробел, а программа сохраняет их в список. Казалось бы, что может пойти не так? На практике здесь кроются 3 ключевые проблемы:

• 🔹 Разделение строки: как правильно разбить ввод по пробелам, если пользователь случайно введёт несколько подряд?
• 🔹 Преобразование типов: почему input() всегда возвращает строку, и как её безопасно конвертировать в int или float?
• 🔹 Обработка ошибок: что делать, если пользователь введёт букву вместо числа?

Минимальный рабочий пример без обработки ошибок:

numbers = list(map(int, input("Введите числа через пробел: ").split()))

Но этот код упадёт, если пользователь введёт "5 a 10". Исправим это с помощью цикла проверки:

while True:
try:
user_input = input("Введите целые числа через пробел: ")
numbers = [int(x) for x in user_input.split()]
break
except ValueError:
print("Ошибка! Введите только числа, разделённые пробелами.")

⚠️ Внимание: Функция split() без аргументов удаляет все пробелы, включая множественные. Если вам нужно сохранить пустые элементы (например, при разборе CSV), используйте split(' ') с явным разделителем.

Обработайте исключение ValueError

Удалите лишние пробелы методом strip()

Предупредите пользователя об ожидаемом формате

Добавьте пример корректного ввода в подсказку-->

2. Ввод массива фиксированной длины: контроль количества элементов

Часто требуется ввести массив с заранее известным количеством элементов (например, координаты точки в 3D-пространстве или оценки студента по 5 предметам). Здесь ключевая сложность — заставить пользователя ввести ровно столько чисел, сколько нужно.

Решение с предварительным указанием размера:

n = int(input("Сколько чисел вы хотите ввести? "))
numbers = []
for i in range(n):
while True:
try:
num = int(input(f"Введите число {i+1}/{n}: "))
numbers.append(num)
break
except ValueError:
print("Это не число! Попробуйте снова.")

Альтернативный подход — ввод всех чисел одной строкой с последующей валидацией:

while True:
user_input = input(f"Введите ровно {n} целых чисел через пробел: ")
numbers = user_input.split()
if len(numbers) != n:
print(f"Ожидалось {n} чисел, получено {len(numbers)}!")
continue
try:
numbers = [int(x) for x in numbers]
break
except ValueError:
print("Все элементы должны быть целыми числами!")

3. Ввод многомерных массивов (матриц): обработка строк и столбцов

Для работы с матрицами или таблицами данных нужно организовать ввод вложенных списков. Здесь главная сложность — согласованность размеров: все строки должны иметь одинаковую длину.

Пример ввода матрицы 3×3:

matrix = []
print("Введите матрицу 3x3 построчно (числа через пробел):")
for i in range(3):
while True:
row_input = input(f"Строка {i+1}: ")
try:
row = list(map(int, row_input.split()))
if len(row) != 3:
print("Должно быть ровно 3 числа!")
continue
matrix.append(row)
break
except ValueError:
print("Ошибка! Только целые числа через пробел.")

Для матриц произвольного размера сначала запрашиваем количество строк и столбцов:

rows = int(input("Количество строк: "))
cols = int(input("Количество столбцов: "))
matrix = []
for i in range(rows):
while True:
row_input = input(f"Строка {i+1} ({cols} чисел): ")
try:
row = list(map(int, row_input.split()))
if len(row) != cols:
print(f"Ожидалось {cols} чисел!")
continue
matrix.append(row)
break
except ValueError:
print("Некорректный ввод!")

⚠️ Внимание: При работе с большими матрицами (100+ элементов) поэлементный ввод неэффективен. Рассмотрите альтернативы: чтение из файла или генерацию данных в коде.

import csv
from io import StringIO

user_input = "1,2,3\n4,5,6\n7,8,9"
matrix = list(csv.reader(StringIO(user_input), delimiter=','))
matrix = [[int(x) for x in row] for row in matrix]

4. Продвинутые техники: использование eval() и его альтернатив

Многие учебники предлагают использовать eval() для быстрого преобразования строки в список:

user_input = input("Введите список в формате [1, 2, 3]: ")
numbers = eval(user_input)

Но этот метод крайне опасен:

• 💣 Выполняет любой Python-код (включая os.system('rm -rf /'))
• 🔓 Нарушает безопасность при работе с пользовательским вводом
• 🐞 Трудно отлаживать ошибки синтаксиса

Безопасные альтернативы:

1. Модуль ast.literal_eval — парсит только литералы (числа, строки, списки, кортежи):

import ast
user_input = input("Введите список в формате [1, 2, 3]: ")
try:
numbers = ast.literal_eval(user_input)
if not isinstance(numbers, list):
raise ValueError("Ожидался список!")
except (ValueError, SyntaxError):
print("Некорректный формат!")

import json
user_input = input("Введите JSON-массив (например, [1, 2, 3]): ")
try:
numbers = json.loads(user_input)
except json.JSONDecodeError:
print("Ошибка разбора JSON!")

def input_array(prompt="Введите числа: ", sep=" ", type=int, min_length=None, max_length=None):
while True:
try:
user_input = input(prompt).strip()
if not user_input:
raise ValueError("Пустой ввод!")
elements = list(map(type, user_input.split(sep)))
if min_length is not None and len(elements) < min_length:
raise ValueError(f"Минимум {min_length} элементов!")
if max_length is not None and len(elements) > max_length:
raise ValueError(f"Максимум {max_length} элементов!")
return elements
except ValueError as e:
print(f"Ошибка: {e}")

# Простой список чисел
nums = input_array()

Список вещественных чисел через запятую
floats = input_array(sep=",", type=float)

Список из 5 элементов
five_nums = input_array(min_length=5, max_length=5)

def input_2d_array(rows, cols, type=int):
array = []
for i in range(rows):
prompt = f"Строка {i+1}/{rows} ({cols} элементов): "
row = input_array(prompt, min_length=cols, max_length=cols, type=type)
array.append(row)
return array

1. Проблема: Пользователь вводит числа с плавающей точкой, а программа ожидает int.
   Решение: Используйте float вместо int или добавьте проверку:

   x = input()
   num = int(float(x)) if '.' in x else int(x)

2. Проблема: Лишние пробелы в начале/конце строки ломают split().
   Решение: Всегда применяйте strip():

   user_input = input().strip()  # Удаляет пробелы с краёв

3. Проблема: Ввод пустой строки вызывает ValueError.
   Решение: Проверяйте длину строки перед преобразованием:

   if not user_input.strip():
   continue  # Пропускаем пустой ввод

numbers = [1, 2, 3]
user_input = input("Добавьте число: ")
numbers.append(user_input)  # Добавляется строка, а не число!

numbers.append(int(user_input))

words = input("Введите слова через пробел: ").split()
или с обработкой пустых элементов:
words = [x for x in input().split() if x]

with open('data.txt') as f:
matrix = [list(map(int, line.split())) for line in f]

# Плохо (если есть лишние пробелы):
numbers = list(map(int, input().split(' ')))

Хорошо:
numbers = list(map(int, input().split()))

while True:
try:
num = int(input("Введите число от 1 до 100: "))
if 1 <= num <= 100:
break
print("Число вне диапазона!")
except ValueError:
print("Это не число!")

if all(1 <= x <= 100 for x in numbers):

import re
user_input = "1, 2, 3; 4,5, 6"
numbers = list(map(int, re.findall(r'\d+', user_input)))
Извлечёт все последовательности цифр: [1, 2, 3, 4, 5, 6]