蟒蛇：讀表中的每一列轉換爲正確的數據類型

Question

假設我有這樣一個文本文件的行象下面這樣：

3, 4.32, hi 
7, 3.23, hello 

當我在這個文件中讀取，所有參賽作品將被視爲字符串。有沒有辦法讓它們自動轉換（在我的例子中）爲int，float和string？

Answer 1

一個原始的方法：

def guess(text): 
    for t in text.split(','): 
     for typ in (int, float, str): 
      try: 
       yield typ(t) 
       break 
      except ValueError as e: 
       pass 

d = '3, 4.32, hi' 

print list(guess(d)) 
# [3, 4.32, ' hi'] 
# [<type 'int'>, <type 'float'>, <type 'str'>] 

但是 - 如果你知道的類型應該是什麼，那麼這比暴力破解這樣好多了......

而且你可以利用ast庫做一些可能更靈活的工作，在try/except機制上需要的命中更少（讀取更容易一點，但需要預處理輸入需要一點點工作）

def guess2(text): 
    from ast import literal_eval 
    tokens = (t.strip() for t in text.split(',')) 
    for token in tokens: 
     try: 
      token_value = literal_eval(token) 
      token_type = type(token_value) 
      yield token_value, token_type 
     except ValueError as e: 
      yield token, type(token) 

Answer 2

如果所有文件中存在固定模式，則使用另一個quickie。

def update_types(): 
    f = open("txtfile", "r") 
    reader = f.readlines() 
    for line in reader: 
     parts = line.split(',') 
     parts[0], parts[1] = int(parts[0]), float(parts[1]) 
     for part in parts: 
      print type(part) 
    f.close() 

Answer 3

如果你事先知道你可以做到這一點的類型：

import csv 
type_funcs = [int, float, str] 
with open('yourfile.csv', 'rb') as f: 
    reader = csv.reader(f) 
    for row in reader: 
     typed_row = [func(val) for func, val in zip(type_funcs, row)]