У меня есть список списков, и я знаю каждый тип элементов [Str, Str, Str, Int, Int, Int, Str, Int]
. У меня есть функция новообращенного, которая угадывает тип:типы преобразования элементов повторного списка
def convert(val):
constructors = [int, str]
for c in constructors:
try:
return c(val)
except ValueError:
pass
Как это можно заменить функцию преобразования, потому что я знаю, тип каждого элемента (смотрите ниже полный код)?
from __future__ import absolute_import, division, print_function
from itertools import groupby
DATA = [["Test", "A", "B01", 828288, 1, 7, 'C', 5],
["Test", "A", "B01", 828288, 1, 7, 'T', 6],
["Test", "A", "B01", 171878, 3, 7, 'C', 5],
["Test", "A", "B01", 171878, 3, 7, 'T', 6],
["Test", "A", "B01", 871963, 3, 9, 'A', 5],
["Test", "A", "B01", 871963, 3, 9, 'G', 6],
["Test", "A", "B01", 1932523, 1, 10, 'T', 4],
["Test", "A", "B01", 1932523, 1, 10, 'A', 5],
["Test", "A", "B01", 1932523, 1, 10, 'X', 6],
["Test", "A", "B01", 667214, 1, 14, 'T', 4],
["Test", "A", "B01", 667214, 1, 14, 'G', 5],
["Test", "A", "B01", 667214, 1, 14, 'G', 6]]
def convert(val):
constructors = [int, str]
for c in constructors:
try:
return c(val)
except ValueError:
pass
def main():
with open("/home/mic/tmp/test.txt") as f:
for line in f:
try:
data = [convert(part.strip()) for part in line.split(',')]
print(data)
except IndexError:
continue
UPDATE Спасибо за все ответы, которые дали мне новые идеи, и поэтому я изменил код (Метод 1 - 4), который в настоящее время не работает:
#!/usr/bin/env python
from __future__ import absolute_import, division, print_function
from itertools import groupby
import csv
parts = [["Test", "A", "B01", 828288, 1, 7, 'C', 5],
["Test", "A", "B01", 828288, 1, 7, 'T', 6],
["Test", "A", "B01", 171878, 3, 7, 'C', 5],
["Test", "A", "B01", 171878, 3, 7, 'T', 6],
["Test", "A", "B01", 871963, 3, 9, 'A', 5],
["Test", "A", "B01", 871963, 3, 9, 'G', 6],
["Test", "A", "B01", 1932523, 1, 10, 'T', 4],
["Test", "A", "B01", 1932523, 1, 10, 'A', 5],
["Test", "A", "B01", 1932523, 1, 10, 'X', 6],
["Test", "A", "B01", 667214, 1, 14, 'T', 4],
["Test", "A", "B01", 667214, 1, 14, 'G', 5],
["Test", "A", "B01", 667214, 1, 14, 'G', 6]]
def iter_something(rows):
key_names = ['type', 'name', 'sub_name', 'pos', 's_type', 'x_type']
chr_key_names = ['letter', 'no']
for keys, group in groupby(rows, lambda row: row[:6]):
result = dict(zip(key_names, keys))
result['chr'] = [dict(zip(chr_key_names, row[6:])) for row in group]
yield result
def main():
#Method 1
converters = [str, str, str, int, int, int, str, int]
with open("/home/mic/tmp/test.txt") as f:
parts = (line.strip().split(',') for line in f)
column = (con(part) for con, part in zip(converters, parts))
for object_ in iter_something(column):
print(object_)
#Method 2
with open("/home/mic/tmp/test.txt") as f:
parts = (line.strip().split(',') for line in f)
parts[3], parts[4], parts[5], parts[7] = int(parts[3]),\
int(parts[4]),\
int(parts[5]),\
int(parts[7])
column = (con(part) for con, part in zip(converters, parts))
for object_ in iter_something(column):
print(object_)
#Method 3
converters = [str, str, str, int, int, int, str, int]
with open("/home/mic/tmp/test.txt", 'rb') as f:
reader = csv.reader(f, skipinitialspace=True)
for object_ in iter_something(reader):
print(object_)
#Method 4
with open("/home/mic/tmp/test.txt", 'rb') as f:
reader = csv.reader(f, skipinitialspace=True)
reader[3], reader[4], reader[5], reader[7] = int(reader[3]),\
int(reader[4]),\
int(reader[5]),\
int(reader[7])
for object_ in iter_something(reader):
print(object_)
if __name__ == '__main__':
main()
что тип каждого элемента и то, что вы хотите конвертировать функция делать? –
converters = [str, str, str, int, int, int, str, int] – user977828
, поэтому вы хотите удалить try и cast, потому что вы знаете, что структура всегда будет str, str, str, int, int, int, str, int? –