回答此问题可获得 20 贡献值,回答如果被采纳可获得 50 分。
<p>我有一个列表,我知道每种类型的元素<code>[Str, Str, Str, Int, Int, Int, Str, Int]</code>。我有一个convert函数,可以猜测类型:</p>
<pre><code>def convert(val):
constructors = [int, str]
for c in constructors:
try:
return c(val)
except ValueError:
pass
</code></pre>
<p>如何替换convert函数,因为我知道每个元素的类型(请参见下面的完整代码)?你知道吗</p>
<pre><code>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
</code></pre>
<p><strong>更新
感谢您的所有回复,这些回复给了我新的想法,因此我也修改了代码(<em>方法1-4</em>),该代码目前不起作用:</p>
<pre><code>#!/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()
</code></pre>