Почему не работает split()?

Question

knownamed @knownamed

Python

Почему не работает split()?

У меня есть задача сделать так, чтобы я мог извлекать столбцы из файла формата:

HETATM 1 H HOH A 1 27.265 36.739 58.126
HETATM 2 H HOH A 1 27.109 35.124 57.944
HETATM 3 O HOH A 1 27.486 35.958 57.542
...
HETATM 9999 O HOH A3333 30.490 83.899 10.929
END

Но, стандартный .split() почему-то не работает и не хочет делить файл по пробелам, работает только с первым столбцом и выдавая ошибку на последующих: IndexError: list index out of range
Код:

water = open('watercube.pdb','r+')
water2 = water.readlines()
for col in water2:
    el = [col.split()[1] for col in water2]

Вопрос задан более трёх лет назад
190 просмотров

2 комментария

Подписаться 1 Простой 2 комментария

o5a @o5a

Если ипользуется генератор списков
el = [col.split()[1] for col in water2]
то уже не нужно запускать цикл
for col in water2:

Как уже заметили, ошибка у Вас потому, что последняя строка не содержит пробелов, поэтому индекса 1 при ее разделении не будет.
Отсекайте эту строку (если явно знаете, какие строки не подходят)
el = [col.split()[1] for col in water2[:-1]]
или добавляйте проверку длины списка или количества пробелов.

Написано более трёх лет назад

devdb @devdb

1) не col, а row или line - правильно называем переменные
2)

first_col = []
for row in water2:
    if not row or not row.strip() or row.strip().lower()=='end':  break  #закончить цикл, если строка пустая или содержит только END
    first_col.append( row.split()[1] )

Написано более трёх лет назад

Если ипользуется генератор списков
el = [col.split()[1] for col in water2]
то уже не нужно запускать цикл
for col in water2:

Как уже заметили, ошибка у Вас потому, что последняя строка не содержит пробелов, поэтому индекса 1 при ее разделении не будет.
Отсекайте эту строку (если явно знаете, какие строки не подходят)
el = [col.split()[1] for col in water2[:-1]]
или добавляйте проверку длины списка или количества пробелов.
1) не col, а row или line - правильно называем переменные
2)

first_col = [] for row in water2: if not row or not row.strip() or row.strip().lower()=='end': break #закончить цикл, если строка пустая или содержит только END first_col.append( row.split()[1] )

Answer 1 · 2020-12-16 16:09:24

Конкретно в вашем случае он скорее всего ругался, когда обрабатывал последнюю строку END. там split возвращает список из одного элемента, а вы обращаетесь ко второму элементу.

Да и for col in water2: тут не нужен.

water2=['HETATM 1 H HOH A 1 27.265 36.739 58.126', 
'HETATM 2 H HOH A 1 27.109 35.124 57.944', 
'HETATM 3 O HOH A 1 27.486 35.958 57.542']
el = [col.split() for col in water2]

Почему не работает split()?

Войдите на сайт