python - 在 python 的数据框中添加 csv 文件中的列

Question

我使用 writer 在 python 中创建了一个 csv 文件。我将此 csv 文件称为“数据”，它由一列 40 行组成。然后我在新的 python 脚本中导入这个 csv 文件并创建一个数据框。然而，在我的新数据框中，我想将这 40 个数据点分成 4 列，每列 10 行。到目前为止我尝试过的方法如下:

    data=pandas.read_csv("location/data.csv", header=None)
    frame=pandas.DataFrame(data[:10])
    frame['second column']=data[10:20]

这会在数据帧“frame”中创建一个新列，称为“第二列”，但所有条目都是 NAN。看来 python 无法识别 csv 文件中的这些条目。任何帮助将不胜感激。下面是我用来创建 csv 文件的代码:

    Var1="data"
    with open(Var1,"ab") as output:
           writer=csv.writer(output, lineterminator='\n')
           for val in variable:
                writer.writerow([val])

这样，我基本上将数组“变量”的值保存在单独的 csv 文件中。

Answer 1

这不是最优雅的解决方案，但我认为它确实解决了您的问题。

import numpy as np
import pandas as pd

x = list(range(0, 40))
data = pd.DataFrame(x, columns=["A"])
arr = data.as_matrix()

data = pd.DataFrame(arr.reshape((10, 4), order="F"),
                    columns=['A', 'B', 'C', 'D'])

要更改重构后的数据中的顺序，只需将order参数从“F”更改为“C”。