如何編寫一個將每行分隔爲逗號字符的函數？

Question

例如：

Myfunction("1,2,3\n4,5,6") 

輸出將是[["1","2","3"],["4","5","6"]]

Answer 1

使用列表理解：

def myfunction(somestring): 
    return [line.split(',') for line in somestring.split('\n')] 

演示：

>>> def myfunction(somestring): 
...  return [line.split(',') for line in somestring.split('\n')] 
... 
>>> myfunction("1,2,3\n4,5,6") 
[['1', '2', '3'], ['4', '5', '6']] 

或者，你可以使用str.splitlines()，這像一樣工作但將分割數據任意換行符組合，即\r,\n或\r\n。它處理最後一行也更聰明一些。

如果此數據來自文件，請考慮使用正確的工具;該csv module可以處理引用上更好的逗號分隔的數據的複雜性：

import csv 

with open('/your/csv/file.csv', 'rb') as inputfile: 
    reader = csv.reader(inputfile) 
    for row in reader: 
     # row is a list of column values 

的數據不有來自一個文件，csv可以處理任何可迭代的，包括.splitlines()結果：

reader = csv.reader(somestring.splitlines()) 
for row in reader: 
    # row is a list of column values 

Answer 2

如果你正在寫這個功能從文件中讀取的東西，那麼csv模塊是你的朋友：

import csv 

with open('somefile.txt') as f: 
    reader = csv.reader(f, delimiter=',') 
    lines = list(reader) 

Answer 3

使用列表理解與str.split和str.splitlines：

>>> strs = "1,2,3\n4,5,6" 
>>> strs.splitlines() 
['1,2,3', '4,5,6'] 
>>> [x.split(',') for x in strs.splitlines()] 
[['1', '2', '3'], ['4', '5', '6']] 

幫助上str.splitlines：

>>> print str.splitlines.__doc__ 
S.splitlines(keepends=False) -> list of strings 

Return a list of the lines in S, breaking at line boundaries. 
Line breaks are not included in the resulting list unless keepends 
is given and true. 

Answer 4

如果輸入的長度爲N M行，這裏有一個發電機需要O（N）的空間而不是O（M * N）：

def myfunction(text): 
    from StringIO import StringIO 
    for line in StringIO(text): 
     yield line.rstrip().split(',') 

它在ord中返回迭代器而不是具體列表呃減少內存消耗。你可以在for循環或類似的地方使用結果，如果你真的需要輸出爲列表，只需將它傳遞給構造函數list()。