从.csv文件拆分数据并绘制图表

Posted 2021-03-27

我有一个问题，我有一个像这样的.csv文件：

Name1
x1,y1
x2,y3
x3,y3
.
.
x100,y100

Name2
x1,z1
x2,z2
x3,z3
.
.
x100,z100

Name3
...
...
Name10
...
...

有什么方法可以在熊猫中分割这些数据，这样我就可以用它绘制图形。

标签将是Name1，Name2 ......

答案

不确定你的图形需要什么样子，但这里有一种方法可以将你的csv文件加载到pandas数据帧中：

import pandas

# declarations
data = []
data_dict = {}

with open('file.csv', 'r') as f:
    while True:
        csv_line = f.readline()
        if not csv_line:
            break
        split_text = csv_line.replace('
', '').split(',')
        if len(split_text) == 1 and split_text != ['']:
            header = split_text[0]
            data = []
            data_dict.update({header: data})
        elif len(split_text) == 2:
            data += split_text

df = pandas.DataFrame(data_dict)