pandas教程01: pandas的安装和基本操作

pandas是Python中常用的数据处理库，主要用来处理表格数据，类似于下面这种：

大家想一想，无论是日常办公使用的excel还是数据库，是不是都是上面这种形式的数据？pandas使用DataFrame统一表示这类表格数据。

DataFrame是什么？这并不重要，把它看作上面的这种表格就好。只要知道在使用pandas时，我们需要将文件里保存的数据转化为DataFrame，所有的操作都是在DataFrame上进行的，最后将处理好的DataFrame保存为文件。大致就是这个流程。

1. 安装和导入pandas

在使用pandas前，我们需要安装它，安装方式很简单，在安装了Python后，以下一行命令：

pip install pandas

就完成了安装。

使用时，首先需要导入pandas库。在Python中我们常通过以下方式导入pandas库:

import pandas as pd

即用别名pd替代pandas, 这已经成为了一种共识。

2. 从文件中导入表格

pandas为常见的文件提供了导入的接口，例如xlsx, csv, sqlite等等。

我们以csv文件为例，新建一个文件"薪资.csv"，用记事本打开后，复制以下内容到此文件中：

姓名,年龄,性别,年薪,奖金
久九刘,35,男,182600,42000
傅儿待,24,男,9960000,40000000
舍处,28,女,60000,18000

通过以下代码，将"薪资.csv"文件里的数据导入为DataFrame:

df = pd.read_csv('薪资.csv')

这样，我们导入"薪资.csv"中的数据为DataFrame, 并保存在变量df中。

如果想导入excel格式的文件，可以写为：df = pd.read_excel('薪资.xlsx')

3. 对DataFrame的基本操作

现在我们看看df是什么，使用print(df)可以看到以下输出：

    姓名  年龄 性别       年薪        奖金
0  久九刘  35  男   182600     42000
1  傅儿待  24  男  9960000  40000000
2   舍处  28  女    60000     18000

可以看到, df就是一张表格。不过请注意，df多了0, 1, 2这一列，这是因为pandas会为表格中的每一行添加索引（索引是从0开始的），之后我们将其保存到文件中时还会再提到这一点。

删除行

删除使用drop方法。如果我们想要删除一行，这时候就使用刚才提到的索引，比如说删除第1行：

df.drop(1)

得到的结果如下：

    姓名  年龄 性别      年薪     奖金
0  久九刘  35  男  182600  42000
2   舍处  28  女   60000  18000

如果想同时删除多行，可以通过列表指定多行。例如同时删除第0、2行：

df.drop([0, 2])

得到的结果如下：

    姓名  年龄 性别       年薪        奖金
1  傅儿待  24  男  9960000  40000000

需要注意的是，如果我们此时查看df, df并没有发生改变。这是因为pandas默认不对原表格处理，会复制一份新的结果返回给用户（想一想，这样是不是可以避免误操作损坏原先的数据？）。如果你想直接在df上操作，可以在drop中指定参数inplace=True(这个参数默认为False).

df.drop([0, 2], inplace=True)

如果此时查看df，就会发现df的第0行和第2行已经被删掉了。

删除列

删除列也是使用drop, 不过需要指定参数axis=1, 表示对列操作(此参数默认为0, 表示对行操作). 例如删除“性别”这一列：

df.drop("性别", axis=1, inplace=True)

这行代码表示在df上直接删除性别这一列。删除多列和删除多行类似，传入一个列表参数即可，例如删除“年龄”和“年薪”这两列：

df.drop(["年龄", "年薪"], axis=1, inplace=True)

4. 保存DataFrame到文件

在我们处理好表格df后，可以将表格保存为xlsx, csv等文件。例如保存到文件"薪资1.csv"中：

df.to_csv('薪资1.csv')

不过，如果此时打开"薪资1.csv"文件，你会发现索引这一列也被保存到文件中。多数情况下，大家并不需要索引这一列，这样在保存时就可以指定参数index=False:

df.to_csv('薪资1.csv', index=False)

此时再打开文件，就会发现索引这一列消失了。

如果想要保存为excel文件，例如保存为"薪资1.xlsx"：

df.to_csv('薪资1.xlsx', index=False)

如果你在使用此函数时遇到错误：ModuleNotFoundError: No module named 'openpyxl'，那么请安装下面这个库：

pip install openpyxl

安装完成后，再重新试下，应该就没有问题了。

5. 小结

恭喜你，已经和pandas成为了朋友，接下来就让我们一起更加深入地了解他吧！

以前写教程，差不多写到第2篇就没有后续了，希望这次能坚持下去。