将.CSV文件导入Python以生成散点图和直方图

Question

我正在尝试将.CSV文件（从Excel文件转换）导入到Python中，这样我就可以制作相关/散点图和直方图。

我该怎么做？

Answer 1

如果您需要逐行使用csv文件，则可以使用csv模块，pandas和matplotlib模块为数据分析任务提供了更高级别的界面。

data.csv

x,y
1,2
2,4
3,6
4,7
5,11
6,12
7,13
8,20
9,17
10,19

plots.py

import pandas as pd
import matplotlib.pyplot as plt
df  = pd.read_csv("data.csv")
df.plot()  # plots all columns against index
df.plot(kind='scatter',x='x',y='y') # scatter plot
df.plot(kind='density')  # estimate density function
# df.plot(kind='hist')  # histogram

输出

如何运作

df  = pd.read_csv("data.csv")

read_csv()将csv文件读入Pandas Dataframe

数据框图方法是matplotlib图的包装，是documented here

请注意，我们可以通过将kind=关键字参数调整为df.plot()来获得不同类型的图表。在matplotlib的较新版本中，直方图可用于kind='hist'

Answer 2

Python内置了对csv文件的支持：https://docs.python.org/2/library/csv.html。文档中有几个例子。

Answer 3

首先import csv，然后您可以使用此代码打开csv文件。 使用行for row in reader:，您可以遍历csv文件中的行，并使用您的代码执行您需要执行的操作。

import csv
with open('your_file.csv', 'rb') as f:
    reader = csv.reader(f)
    for row in reader:
        #do something