從使用熊貓的行列表中加載數據

Question

我想從文本行列表中加載一個熊貓數據框，很像使用pa.read_csv（）函數。

所以，如果我有一個文本文件是這樣的：

col1 col2 
1 2 
3 4 

我可以加載到像這樣的熊貓數據幀：

>>> import pandas as pa 
>>> pa.read_csv('test.txt') 
    col1 col2 
0  1 2 
1  3 4 

我的問題是，我可以加載像數組這個：

>>> lines = ['col1 col2', '1 2', '3 4'] 

並使用一些函數來加載，作爲一個數據幀，在文件的情況下？

Answer 1

試試這個：

import StringIO 

pa.read_csv(StringIO.StringIO(lines.join('\n'))) 

Answer 2

我懷疑你需要大熊貓做到這一點，你可以這樣做只是普通的Python：

with open('data.txt', 'r') as f: 
    lines = [l.rstrip() for l in f.readlines()] 

print(lines) 

輸出：

['col1 col2', '1 2', '3 4']