从具有相同列的一个 CSV 文件中合并 Pandas 数据框

Question

我希望我的问题不是另一个问题的重复，但我已经搜索了三天，但还没有找到答案。

好的，我有一个包含两个标题的 CSV 文件。该文件包含有关酒店（他们的名称）、他们的成本（价格）、他们的评级以及他们的位置（区域 1、2 或 3）的信息：

The CSV file imported

如您所见，第一行描述了该区域，而第二行是酒店名称、价格和评级。我想要的是重新排列文件并将其保存到一个新的 CSV 文件中，格式如下所示：

The hopeful output

因此，有关酒店区域的信息已被赋予了自己的列。第二行的名字都是一样的。 有没有办法创建这个？当必须导入这些树状数据结构时，我有点陌生。如果树有更多节点（例如，如果我们从国家开始，向下移动到区域，然后向下移动到酒店名称、价格和评级），是否可以完成？可以用 Pandas 完成吗？

Answer 1

首先，您可以将 csv 文件作为文本文件共享吗？这对尝试我自己的解决方案非常有帮助。把图片中的数据写下来似乎效率不高。

其次，您是否尝试过通过自己编写脚本来实现这一目标？或者你有没有试过使用一些图书馆？您添加了标签pandas，但在文本中您没有提及。它应该是熊猫有什么具体原因吗？

仅通过使用切片就可以轻松实现适用于该情况的解决方案。我猜你的格式是相当具体的而不是标准的，所以这些库可能没有多大帮助。熊猫允许多行作为标题，但以不同的方式解释，请参阅pandas dataframe with 2-rows header and export to csv

一个解决方案：

table = []
with open(my_csv_file) as f:
    for line in f:
        a1, p1, r1, a2, p2, r2, a3, p3, r3 = line[:-1].split(",")
        table.append([a1, p1, r1, "area1"])
        table.append([a2, p2, r2, "area2"])
        table.append([a3, p3, r3, "area3"])
# ... convert table into dataframe etc.

Answer 2

好的，所以我为这个问题创建了一个可能的解决方案：

infile = csv.reader(infile, delimiter=';')
    out = []
    counter = 0
    i = 0
    k = 0
    names = []
    temp1 = 0
    for line in infile:
        temp = list(set(line))
        if counter == 0:
            names = line
            counter +=1
        elif counter == 1:
            k = len(list(set(line)))
            while i < len(line):
                line.insert(i+k, name)
                i += (k + 1)
            counter += 1
            out.append(line)
        else:
            i = 0
            ind = 0
            while i < len(line):
                line.insert(i+k, names[ind*k])
                i += (k + 1)
                ind +=1
            out.append(line)
    headers = out.pop(0)
    n = len(set(headers))
    table = pd.DataFrame(out, columns=headers)
    for i in range(0, len(table.columns)):
        if i ==0:
            temp1 = table.ix[:,n*i:n*(i+1)]
        else:
            temp1 = pd.concat([temp1, table.ix[:,n*i:n*(i+1)]], ignore_index=True)

我非常想要一些输入和建议，以使解决方案更优雅或向文件添加额外级别的标题。