使用python读取每个CSV文件列(最好是通过pandas的帮助)

时间:2013-11-06 15:43:20

标签: python excel csv

我在Microsoft Excel中有一些数据,我将它们保存为CSV文件以方便使用。数据结构是这样的:

MS Excel格式:

L1                      
0   1   0   0   0   1   1
0   0   1   0   0   1   0
0   0   0   1   0   0   1
0   0   0   0   1   0   0
1   1   1   1   1   1   1
1   1   1   1   1   1   1
1   1   1   1   1   1   1
1   1   1   1   1   1   1

CSV格式

L1,,,,,,,,,,,,,,
0,1,0,0,0,1,1,
0,0,1,0,0,1,0,
0,0,0,1,0,0,1,
0,0,0,0,1,0,0,
1,1,1,1,1,1,1,
1,1,1,1,1,1,1,
1,1,1,1,1,1,1,
1,1,1,1,1,1,1,

正如您所看到的,现在只有第一列有标签我想读取CSV文件(或者excel文件更容易)获取每一列并对它们进行一些位操作操作。我该如何实现这一目标?我已经阅读了一些关于熊猫的内容但是我找不到任何有用的东西来获取每个coloumn

4 个答案:

答案 0 :(得分:2)

给出.csv文件temp.csv

L1x,,,,,,,
0,1,0,0,0,1,1,
0,0,1,0,0,1,0,
0,0,0,1,0,0,1,
0,0,0,0,1,0,0,
1,1,1,1,1,1,1,
1,1,1,1,1,1,1,
1,1,1,1,1,1,1,
1,1,1,1,1,1,1,

按如下方式阅读:

import pandas
a = pandas.read_csv('temp.csv', names = ["c%d" % i for i in range(8)], skiprows = 1)
a

输出:

   c0  c1  c2  c3  c4  c5  c6  c7
0   0   1   0   0   0   1   1 NaN
1   0   0   1   0   0   1   0 NaN
2   0   0   0   1   0   0   1 NaN
3   0   0   0   0   1   0   0 NaN
4   1   1   1   1   1   1   1 NaN
5   1   1   1   1   1   1   1 NaN
6   1   1   1   1   1   1   1 NaN
7   1   1   1   1   1   1   1 NaN

最后一栏中的“NaN's”来自讨厌的尾随逗号。范围中的8需要与列数匹配。要访问a中的列,请使用

a.c3

a[c3]

两者都导致

0    0
1    0
2    1
3    0
4    1
5    1
6    1
7    1
Name: c3

关于熊猫的一件很酷的事情是,如果你想要对两列进行异或,你可以非常简单。

a.c0^a.c2

输出

0    0
1    1
2    0
3    0
4    0
5    0
6    0
7    0
Name: c0

答案 1 :(得分:1)

假设我有:

enter image description here

您可以将其保存为看起来像这样的CSV文件:

L1,,,
L2,0,10,20
L3,1,11,21
L4,2,12,22
L5,3,13,23
L6,4,14,24
L7,5,15,25
L8,6,16,26
L9,7,17,27
L10,8,18,28

要获得任何col,请使用CSV阅读器并使用zip:

进行转置
import csv

with open('test.csv', 'rU') as fin:
    reader=csv.reader(fin)
    data=list(reader)

print 'data:', data
# data: [['L1', '', '', ''], ['L2', '0', '10', '20'], ['L3', '1', '11', '21'], ['L4', '2', '12', '22'], ['L5', '3', '13', '23'], ['L6', '4', '14', '24'], ['L7', '5', '15', '25'], ['L8', '6', '16', '26'], ['L9', '7', '17', '27'], ['L10', '8', '18', '28']]

请注意,数据是行列表。您可以使用zip转置该列表列表以获取列列表:

trans=zip(*data)
print 'trans:',trans
# trans: [('L1', 'L2', 'L3', 'L4', 'L5', 'L6', 'L7', 'L8', 'L9', 'L10'), ('', '0', '1', '2', '3', '4', '5', '6', '7', '8'), ('', '10', '11', '12', '13', '14', '15', '16', '17', '18'), ('', '20', '21', '22', '23', '24', '25', '26', '27', '28')]

然后只需索引以获取特定列:

print trans[0]
# ('L1', 'L2', 'L3', 'L4', 'L5', 'L6', 'L7', 'L8', 'L9', 'L10')

当然,如果您想对单元格进行算术运算,则需要根据需要将字符串转换为整数或浮点数。

答案 2 :(得分:0)

import pandas as pd
pd.read_excel("foo.xls", "Sheet 1",
              names=["c%d" % i for i in range(7)])

输出:

   c0  c1  c2  c3  c4  c5  c6
0   0   1   0   0   0   1   1
1   0   0   1   0   0   1   0
2   0   0   0   1   0   0   1
3   0   0   0   0   1   0   0
4   1   1   1   1   1   1   1
5   1   1   1   1   1   1   1
6   1   1   1   1   1   1   1
7   1   1   1   1   1   1   1

答案 3 :(得分:0)

示例代码将列作为数组返回。:

input = """L1,,,,,,,,,,,,,,
0,1,0,0,0,1,1,
0,0,1,0,0,1,0,
0,0,0,1,0,0,1,
0,0,0,0,1,0,0,
1,1,1,1,1,1,1,
1,1,1,1,1,1,1,
1,1,1,1,1,1,1,
1,1,1,1,1,1,1,
"""

def getColumn(data,column_number):

    dump_array=[]
    lines=data.split("\n")
    for line in lines:
        tmp_cell = line.split(",")
        dump_array.append(tmp_cell[3])
    return dump_array

#for ex. get column 3
getColumn(3,input)

这可能会给你的网格制作一个想法...

注意:我现在没有代码来测试代码,如果有拼写错误就很抱歉......