0

我的 csv 文件分布在多个目录中,每个 csv 文件只有一列包含数据。我想要做的是读取所有这些文件并将每个文件的列放入 csv 文件中。最终的 csv 文件将以文件名作为其标题的列,并将其原始文件中的相应数据作为其列数据。

这是我里面的目录结构~/csv_files/ ls

ab   arc  bat-smg   bn       cdo  crh      diq  es   fo   gd   haw  ia   iu   ki   ksh  lez  lv   mo   na      no   os   pih  rmy   sah  simple  ss   tet  tr   ur   war  zea
ace  arz  bcl       bo       ce   cs       dsb  et   fr   gl   he   id   ja   kk   ku   lg   map-bms  mr   nah     nov  pa   pl   rn    sc   sk      st   tg   ts   uz   wo   zh
af   as

每个目录都有两个 csv 文件,我想使用 os.walk() 函数,但我认为我对 os.walk 的理解是不正确的,这就是为什么目前我所拥有的没有产生任何东西。

import sys, os
import csv

root_path = os.path.expanduser(
    '~/data/missing_files')

def combine_csv_files(path):
    for root, dirs, files in os.walk(path):
        for dir in dirs:
            for name in files:
                if name.endswith(".csv"):
                    csv_path = os.path.expanduser(root_path + name)
                    if os.path.exists(csv_path):
                        try:
                            with open(csv_path, 'rb') as f:
                                t = f.read().splitlines()
                                print t
                        except IOError, e:
                            print e

def main():
    combine_csv_files(root_path)

if __name__=="__main__":
    main()

我的问题是:

  1. 我在这里做错了什么?
  2. 我可以从另一个文件中读取一个 csv 列并将该数据作为列添加到另一个文件,因为 csv 文件更依赖于行,并且行之间没有依赖关系。

最后我试图得到这样的csv文件,(这里是潜在的标题)

ab_csv_data_file1, ab_csv_data_file2, arc_csv_data_file1, arc_csv_data_file2
4

2 回答 2

2

您错误地使用 os.walk()

def combine_csv_files(path):
    for root, dirs, files in os.walk(path):
        for name in files:
            if name.endswith(".csv"):
                csv_path = os.path.join(root, name)
                try:
                    with open(csv_path, 'rb') as f:
                        t = f.read().splitlines()
                        print t
                except IOError, e:
                    print e

os.walk() 函数产生一个 3 元组(目录路径、目录名、文件名)。其中“dirpath”是当前行走目录的路径,“dirnames”是“dirpath”中的目录列表,“filenames”是“dirpath”中的文件列表。“dirpath”可能是此处的“路径”,以及“路径”的任何子文件夹。

于 2013-04-16T21:30:54.733 回答
1

我不知道我是否明白你的意思。让你有多个文件夹,如“ab”、“arc”等。对于每个文件夹,它包含两个 CSV 文件。

如果我是对的,那么你没有做正确的事。

def combine_csv_files(path):
    for root, dirs, files in os.walk(path):
        for dir in dirs:
            for dirpath, sub_dirs, sub_files in os.walk('/'.join([path,dir])
                for name in sub_files:
                    if name.endswith(".csv"):
                        csv_path = os.path.expanduser(dirpath + name)
                        if os.path.exists(csv_path):
                            try:
                                with open(csv_path, 'rb') as f:
                                    t = f.read().splitlines()
                                    print t
                            except IOError, e:
                                print e

如果我是对的,上面的代码应该可以工作

于 2013-04-16T21:40:25.680 回答