每列读取CSV文件

Question

我有一个包含四列的数据集，其中前三列是位置参数XYZ，后者是该位置的某个值。 我想以一种特定的方式将此文件读入np.array ，即对于每个点数xnodes我想从最后一列中读取多个值（假设为10）。 我到目前为止所拥有的是：

with open(FilePath, 'rt') as f:

    reader = csv.reader((row for row in f if not row.startswith('#')), skipinitialspace=True, delimiter=' ')

            for i, row in enumerate(reader):

                if i % xnodes == 0:

其中FilePath是数据文件的位置。 现在，当满足最后一个if条件时，我想存储从索引i开始到数组的最后一列的10个值。

由于我对Python的了解有限，所以我不知道该如何实现。 谁能帮我？

谢谢！

Answer 1

就像我在评论中说的，您应该使用Pandas库处理表：

import pandas as pd

""" your if your csv contains this :
       x    y   z   value
       10   0   0    125
       0   10   0    354
       12  21  32    214"""

df = pd.read_csv('example.csv') # by default the separator is a comma

for item in df.index:
   # do stuff with your column
   print(df.loc[item])

输出：

x  10
y  0
z  12
value  125
Name: 0, dtype: int64 

x   0
y   10 
z   21
value   214
Name: 1, dtype: int64
etc...