学习Pandas,第 3 课

英文原文: 03 - Lesson

获取数据 - 我们的数据在一个 Excel 文件中,包含了每一个日期的客户数量。 我们将学习如何读取 Excel 文件的内容并处理其中的数据。
准备数据 - 这组时间序列的数据并不规整而且有重复。 我们的挑战是整理这些数据并且预测下一个年度的客户数。
分析数据 - 我们将使用图形来查看趋势情况和离群点。我们会使用一些内置的计算工具来预测下一年度的客户数。
表现数据 - 结果将会被绘制成图形。

注意: 确保你已经看过了之前的课程,这里的一些练习会需要你在之前课程学到的那些知识。

  1. # 导入所需要的库
  2. import pandas as pd
  3. import matplotlib.pyplot as plt
  4. import numpy.random as np
  5. import sys
  6. import matplotlib
  7. %matplotlib inline
  1. print('Python version ' + sys.version)
  2. print('Pandas version: ' + pd.__version__)
  3. print('Matplotlib version ' + matplotlib.__version__)
  1. Python version 3.6.1 | packaged by conda-forge | (default, Mar 23 2017, 21:57:00)
  2. [GCC 4.2.1 Compatible Apple LLVM 6.1.0 (clang-602.0.53)]
  3. Pandas version: 0.19.2
  4. Matplotlib version 2.0.2

我们将创建一些测试数据用来分析

  1. # 设置种子
  2. np.seed(111)
  3. # 生成测试数据的函数
  4. def CreateDataSet(Number=1):
  5. Output = []
  6. for i in range(Number):
  7. # 创建一个按周计算的日期范围(每周一起始)
  8. rng = pd.date_range(start='1/1/2009', end='12/31/2012', freq='W-MON')
  9. # 创建一些随机数
  10. data = np.randint(low=25, high=1000, size=len(rng))
  11. # 状态池
  12. status = [1, 2, 3]
  13. # 创建一个随机的状态列表
  14. random_status = [status[np.randint(low=0, high=len(status))] for i in range(len(rng))]
  15. # 行政州(state)的列表
  16. states = ['GA','FL','fl','NY','NJ','TX']
  17. # 创建一个行政周的随机列表
  18. random_states = [states[np.randint(low=0, high=len(states))] for i in range(len(rng))]
  19. Output.extend(zip(random_states, random_status, data, rng))
  20. return Output

现在我们有了一个生成测试数据的函数,我们来创建一些数据并放到一个 dataframe 中。

  1. dataset = CreateDataSet(4)
  2. df = pd.DataFrame(data=dataset, columns=['State','Status','CustomerCount','StatusDate'])
  3. df.info()
  1. <class 'pandas.core.frame.DataFrame'>
  2. RangeIndex: 836 entries, 0 to 835
  3. Data columns (total 4 columns):
  4. State 836 non-null object
  5. Status 836 non-null int64
  6. CustomerCount 836 non-null int64
  7. StatusDate 836 non-null datetime64[ns]
  8. dtypes: datetime64[ns](1), int64(2), object(1)
  9. memory usage: 26.2+ KB
  1. df.head()
















































State Status CustomerCount StatusDate
0 GA 1 877 2009-01-05
1 FL 1 901 2009-01-12
2 fl 3 749 2009-01-19
3 FL 3 111 2009-01-26
4 GA 1 300 2009-02-02

现在我们将这个 dataframe 保存到 Excel 文件中,然后再读取出来放回到 dataframe 中。 我们简单地展示一下如何读写 Excel 文件。

我们不会把索引值(index)写到 Excel 文件中,这些索引值不是我们的测试数据的一部分。

  1. # 结果保存到 Excel 中。 译者注: 需要 openpyxl 包
  2. df.to_excel('Lesson3.xlsx', index=False) #不保存索引,但是保存列名(column header)
  3. print('Done')
  1. Done

从 Excel 中获取数据

我们用 read_excel 这个函数从 Excel 文件读取数据。 这个函数允许按照页签的名字或者位置来选择特定的页签(译者注: 即Excel中的sheet)。

  1. pd.read_excel?

注意: 除非指定目录,Excel 文件从与 notebook 相同的目录读取。*

  1. # 文件的位置
  2. Location = r'./Lesson3.xlsx'
  3. # 读取第一个页签(sheet),并指定索引列是 StatusDate
  4. df = pd.read_excel(Location, sheetname=0, index_col='StatusDate') #译者注: 需要 xlrd 包
  5. df.dtypes
  1. State object
  2. Status int64
  3. CustomerCount int64
  4. dtype: object
  1. df.index
  1. DatetimeIndex(['2009-01-05', '2009-01-12', '2009-01-19', '2009-01-26',
  2. '2009-02-02', '2009-02-09', '2009-02-16', '2009-02-23',
  3. '2009-03-02', '2009-03-09',
  4. ...
  5. '2012-10-29', '2012-11-05', '2012-11-12', '2012-11-19',
  6. '2012-11-26', '2012-12-03', '2012-12-10', '2012-12-17',
  7. '2012-12-24', '2012-12-31'],
  8. dtype='datetime64[ns]', name='StatusDate', length=836, freq=None)
  1. df.head()
















































State Status CustomerCount
StatusDate
2009-01-05 GA 1 877
2009-01-12 FL 1 901
2009-01-19 fl 3 749
2009-01-26 FL 3 111
2009-02-02 GA 1 300

准备数据

这一部分,我们尝试将数据进行清洗以备分析:

  1. 确保 state 列都是大写
  2. 只选择 Status = 1 的那些记录
  3. 对 State 列中的 NJ 和 NY,都合并为 NY
  4. 去除一些离群中 (数据集中一些特别奇异的结果)

让我们先快速看一下 State 列中的大小写情况。

  1. df['State'].unique()
  1. array(['GA', 'FL', 'fl', 'TX', 'NY', 'NJ'], dtype=object)

我们用 upper() 函数和 dataframe 的 apply 属性将 State 的值都转换为大写。 lambda 函数简单地将upper()函数应用到 State 列中的每一个值上。

  1. # 清洗 State 列,全部转换为大写
  2. df['State'] = df.State.apply(lambda x: x.upper())
  1. df['State'].unique()
  1. array(['GA', 'FL', 'TX', 'NY', 'NJ'], dtype=object)
  1. # 只保留 Status == 1
  2. mask = df['Status'] == 1
  3. df = df[mask]

NJ 转换为 NY,仅需简单地:

[df.State == ‘NJ’] - 找出 State 列是 NJ 的所有记录。
df.State[df.State == ‘NJ’] = ‘NY’ - 对 State 列是 NJ 的所有记录,将其替换为 NY

  1. # 将 NJ 转换为 NY
  2. mask = df.State == 'NJ'
  3. df['State'][mask] = 'NY'

现在我们看一下,我们有了一个更加干净的数据集了。

  1. df['State'].unique()
  1. array(['GA', 'FL', 'NY', 'TX'], dtype=object)

这是也许我们可以将数据绘制成图形查看一下数据中是否有任何离群值(outliers)或者不一致(inconsistencies)。 我们使用 dataframe 中的 plot() 函数。

从下图你可以看到图形没有说明什么,这也许是一个信号我们需要做更多的数据处理。

  1. df['CustomerCount'].plot(figsize=(15,5));

png

如果我们看这些数据,我们会意识到同一个 State, StatusDate 和 Status 的组合会有多个值。 这可能意味着我们在处理的数据是脏数据/坏数据/不精确的数据(dirty/bad/inaccurate),但我们不这样假设。 我们假设这个数据集是一个更大更大数据集的一个子集(subset),并且如果我们简单的将 State, StatusDate 和 Status 组合下的 CustomerCount 累加起来, 我们将得到每一天的 全部客户数量 (Total Customer Count)。

  1. sortdf = df[df['State']=='NY'].sort_index(axis=0)
  2. sortdf.head(10)














































































State Status CustomerCount
StatusDate
2009-01-19 NY 1 522
2009-02-23 NY 1 710
2009-03-09 NY 1 992
2009-03-16 NY 1 355
2009-03-23 NY 1 728
2009-03-30 NY 1 863
2009-04-13 NY 1 520
2009-04-20 NY 1 820
2009-04-20 NY 1 937
2009-04-27 NY 1 447

我们的任务是创建一个新的 dataframe,然后对数据进行压缩处理,是的每一个 State 和 StatusDate 组合代表一天的客户数量。 我们可以忽略 Status 列因为这一列我们之前处理过只有 1 这个值了。 要完成这个操作,我们使用 dataframe 的 groupyby()sum() 这两个函数。

注意,我们要使用 reset_index。 如果我们不这么做,我们将无法同时用 State 和 StatusDate 这两列来做分组,因为 groupby 函数需要列(columns)来做为输入(译者注: StatusDate 目前是 index,不是 column)。 reset_index 函数将把 dataframen 中作为索引(index)的 StatusDate 变回普通的列。

  1. # 先 reset_index,然后按照 State 和 StatusDate 来做分组 (groupby)
  2. Daily = df.reset_index().groupby(['State','StatusDate']).sum()
  3. Daily.head()












































Status CustomerCount
State StatusDate
FL 2009-01-12 1 901
2009-02-02 1 653
2009-03-23 1 752
2009-04-06 2 1086
2009-06-08 1 649

Daily 这个 dataframe 中,StateStatusDate 这两列被自动设置为了索引(index)。 你可以将 index 设想为数据库表中的逐渐(primary key),只不过没有唯一性(unique)的限制。 索引中的这些列让我们更容易的可以选择,绘图和执行一些计算。

接下去我们将 Status 删掉,它的值就是 1,没有多少用途了。

  1. del Daily['Status']
  2. Daily.head()





































CustomerCount
State StatusDate
FL 2009-01-12 901
2009-02-02 653
2009-03-23 752
2009-04-06 1086
2009-06-08 649

  1. # 看一下 dataframe 中的索引(index)
  2. Daily.index
  1. MultiIndex(levels=[['FL', 'GA', 'NY', 'TX'], [2009-01-05 00:00:00, 2009-01-12 00:00:00, 2009-01-19 00:00:00, 2009-02-02 00:00:00, 2009-02-23 00:00:00, 2009-03-09 00:00:00, 2009-03-16 00:00:00, 2009-03-23 00:00:00, 2009-03-30 00:00:00, 2009-04-06 00:00:00, 2009-04-13 00:00:00, 2009-04-20 00:00:00, 2009-04-27 00:00:00, 2009-05-04 00:00:00, 2009-05-11 00:00:00, 2009-05-18 00:00:00, 2009-05-25 00:00:00, 2009-06-08 00:00:00, 2009-06-22 00:00:00, 2009-07-06 00:00:00, 2009-07-13 00:00:00, 2009-07-20 00:00:00, 2009-07-27 00:00:00, 2009-08-10 00:00:00, 2009-08-17 00:00:00, 2009-08-24 00:00:00, 2009-08-31 00:00:00, 2009-09-07 00:00:00, 2009-09-14 00:00:00, 2009-09-21 00:00:00, 2009-09-28 00:00:00, 2009-10-05 00:00:00, 2009-10-12 00:00:00, 2009-10-19 00:00:00, 2009-10-26 00:00:00, 2009-11-02 00:00:00, 2009-11-23 00:00:00, 2009-11-30 00:00:00, 2009-12-07 00:00:00, 2009-12-14 00:00:00, 2010-01-04 00:00:00, 2010-01-11 00:00:00, 2010-01-18 00:00:00, 2010-01-25 00:00:00, 2010-02-08 00:00:00, 2010-02-15 00:00:00, 2010-02-22 00:00:00, 2010-03-01 00:00:00, 2010-03-08 00:00:00, 2010-03-15 00:00:00, 2010-04-05 00:00:00, 2010-04-12 00:00:00, 2010-04-26 00:00:00, 2010-05-03 00:00:00, 2010-05-10 00:00:00, 2010-05-17 00:00:00, 2010-05-24 00:00:00, 2010-05-31 00:00:00, 2010-06-14 00:00:00, 2010-06-28 00:00:00, 2010-07-05 00:00:00, 2010-07-19 00:00:00, 2010-07-26 00:00:00, 2010-08-02 00:00:00, 2010-08-09 00:00:00, 2010-08-16 00:00:00, 2010-08-30 00:00:00, 2010-09-06 00:00:00, 2010-09-13 00:00:00, 2010-09-20 00:00:00, 2010-09-27 00:00:00, 2010-10-04 00:00:00, 2010-10-11 00:00:00, 2010-10-18 00:00:00, 2010-10-25 00:00:00, 2010-11-01 00:00:00, 2010-11-08 00:00:00, 2010-11-15 00:00:00, 2010-11-29 00:00:00, 2010-12-20 00:00:00, 2011-01-03 00:00:00, 2011-01-10 00:00:00, 2011-01-17 00:00:00, 2011-02-07 00:00:00, 2011-02-14 00:00:00, 2011-02-21 00:00:00, 2011-02-28 00:00:00, 2011-03-07 00:00:00, 2011-03-14 00:00:00, 2011-03-21 00:00:00, 2011-03-28 00:00:00, 2011-04-04 00:00:00, 2011-04-18 00:00:00, 2011-04-25 00:00:00, 2011-05-02 00:00:00, 2011-05-09 00:00:00, 2011-05-16 00:00:00, 2011-05-23 00:00:00, 2011-05-30 00:00:00, 2011-06-06 00:00:00, 2011-06-20 00:00:00, 2011-06-27 00:00:00, 2011-07-04 00:00:00, 2011-07-11 00:00:00, 2011-07-25 00:00:00, 2011-08-01 00:00:00, 2011-08-08 00:00:00, 2011-08-15 00:00:00, 2011-08-29 00:00:00, 2011-09-05 00:00:00, 2011-09-12 00:00:00, 2011-09-26 00:00:00, 2011-10-03 00:00:00, 2011-10-24 00:00:00, 2011-10-31 00:00:00, 2011-11-07 00:00:00, 2011-11-14 00:00:00, 2011-11-28 00:00:00, 2011-12-05 00:00:00, 2011-12-12 00:00:00, 2011-12-19 00:00:00, 2011-12-26 00:00:00, 2012-01-02 00:00:00, 2012-01-09 00:00:00, 2012-01-16 00:00:00, 2012-02-06 00:00:00, 2012-02-13 00:00:00, 2012-02-20 00:00:00, 2012-02-27 00:00:00, 2012-03-05 00:00:00, 2012-03-12 00:00:00, 2012-03-19 00:00:00, 2012-04-02 00:00:00, 2012-04-09 00:00:00, 2012-04-23 00:00:00, 2012-04-30 00:00:00, 2012-05-07 00:00:00, 2012-05-14 00:00:00, 2012-05-28 00:00:00, 2012-06-04 00:00:00, 2012-06-18 00:00:00, 2012-07-02 00:00:00, 2012-07-09 00:00:00, 2012-07-16 00:00:00, 2012-07-30 00:00:00, 2012-08-06 00:00:00, 2012-08-20 00:00:00, 2012-08-27 00:00:00, 2012-09-03 00:00:00, 2012-09-10 00:00:00, 2012-09-17 00:00:00, 2012-09-24 00:00:00, 2012-10-01 00:00:00, 2012-10-08 00:00:00, 2012-10-22 00:00:00, 2012-10-29 00:00:00, 2012-11-05 00:00:00, 2012-11-12 00:00:00, 2012-11-19 00:00:00, 2012-11-26 00:00:00, 2012-12-10 00:00:00]],
  2. labels=[[0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3], [1, 3, 7, 9, 17, 19, 20, 21, 23, 25, 27, 28, 29, 30, 31, 35, 38, 40, 41, 44, 45, 46, 47, 48, 49, 52, 54, 56, 57, 59, 60, 62, 66, 68, 69, 70, 71, 72, 75, 76, 77, 78, 79, 85, 88, 89, 92, 96, 97, 99, 100, 101, 103, 104, 105, 108, 109, 110, 112, 114, 115, 117, 118, 119, 125, 126, 127, 128, 129, 131, 133, 134, 135, 136, 137, 140, 146, 150, 151, 152, 153, 157, 0, 3, 7, 22, 23, 24, 27, 28, 34, 37, 42, 47, 50, 55, 58, 66, 67, 69, 71, 73, 74, 75, 79, 82, 83, 84, 85, 91, 93, 95, 97, 106, 110, 120, 124, 125, 126, 127, 132, 133, 139, 143, 158, 159, 160, 2, 4, 5, 6, 7, 8, 10, 11, 12, 13, 14, 16, 19, 21, 22, 24, 26, 28, 29, 30, 31, 32, 33, 36, 39, 40, 42, 43, 51, 56, 61, 62, 63, 66, 67, 70, 71, 72, 73, 75, 78, 80, 81, 82, 83, 86, 87, 90, 91, 92, 94, 101, 102, 103, 105, 107, 108, 111, 113, 116, 118, 122, 125, 129, 130, 131, 132, 138, 139, 141, 142, 143, 144, 148, 149, 154, 156, 159, 160, 15, 16, 17, 18, 45, 47, 50, 53, 57, 61, 64, 65, 68, 84, 88, 94, 98, 107, 110, 112, 115, 121, 122, 123, 128, 130, 134, 135, 145, 146, 147, 148, 155]],
  3. names=['State', 'StatusDate'])
  1. # 选择 State 这个索引
  2. Daily.index.levels[0]
  1. Index(['FL', 'GA', 'NY', 'TX'], dtype='object', name='State')
  1. # 选择 StatusDate 这个索引
  2. Daily.index.levels[1]
  1. DatetimeIndex(['2009-01-05', '2009-01-12', '2009-01-19', '2009-02-02',
  2. '2009-02-23', '2009-03-09', '2009-03-16', '2009-03-23',
  3. '2009-03-30', '2009-04-06',
  4. ...
  5. '2012-09-24', '2012-10-01', '2012-10-08', '2012-10-22',
  6. '2012-10-29', '2012-11-05', '2012-11-12', '2012-11-19',
  7. '2012-11-26', '2012-12-10'],
  8. dtype='datetime64[ns]', name='StatusDate', length=161, freq=None)

我们按照每一个州来绘制一下图表。

正如你所看到的,将图表按照不同的 State 区分开,我们能看到更清晰的数据。 你能看到任何离群值(outlier)吗?

  1. Daily.loc['FL'].plot()
  2. Daily.loc['GA'].plot()
  3. Daily.loc['NY'].plot()
  4. Daily.loc['TX'].plot();

png

png

png

png

我们也可以指定一个日期,比如 2012,来绘制图表。
We can also just plot the data on a specific date, like 2012. 我们能够清晰地看到这些州的数据分布很广。 因为这些数据包含了每周的客户数量,数据的变化情况看上去是可以预测的。 在这个教程里面,我们假设没有坏数据并继续往下。

  1. Daily.loc['FL']['2012':].plot()
  2. Daily.loc['GA']['2012':].plot()
  3. Daily.loc['NY']['2012':].plot()
  4. Daily.loc['TX']['2012':].plot();

png

png

png

png

我们假设每个月的客户数量应该是保持相对稳定的。 在一个月内任何在这个特定范围之外的数据都可以从数据集中移除。 最终的结果应该更加的平滑并且图形不会有尖刺。

StateYearMonth - 这里我们通过 State, StatusDate 中的年份(year)和月份(Month) 来分组。
Daily[‘Outlier’] - 一个布尔(boolean)变量值 (True 或者 False),从而我们会知道 CustomerCount 值是否在一个可接受的范围之内。

我们将会用到 transform 而不是 apply。 原因是, transform 将会保持 dataframe 矩阵的形状(shape)(就是行列数不变)而 apply 会改变矩阵的形状。 看过前面的图形我们意识到这些图形不是高斯分布的(gaussian distribution),这意味着我们不能使用均值(mean)和标准差(stDev)这些统计量。 我们将使用百分位数(percentile)。 请注意这里也会有把好数据消除掉的风险。

  1. # 计算离群值
  2. StateYearMonth = Daily.groupby([Daily.index.get_level_values(0), Daily.index.get_level_values(1).year, Daily.index.get_level_values(1).month])
  3. Daily['Lower'] = StateYearMonth['CustomerCount'].transform( lambda x: x.quantile(q=.25) - (1.5*x.quantile(q=.75)-x.quantile(q=.25)) )
  4. Daily['Upper'] = StateYearMonth['CustomerCount'].transform( lambda x: x.quantile(q=.75) + (1.5*x.quantile(q=.75)-x.quantile(q=.25)) )
  5. Daily['Outlier'] = (Daily['CustomerCount'] < Daily['Lower']) | (Daily['CustomerCount'] > Daily['Upper'])
  6. # 移除离群值
  7. Daily = Daily[Daily['Outlier'] == False]

Daily 这个 dataframe 按照每天来汇总了客户数量。 而原始的数据则可能每一天会有多个记录。 我们现在保留下一个用 State 和 StatusDate 来做索引的数据集。 Outlier 列如果是 False 的化代表这条记录不是一个离群值。

  1. Daily.head()


























































CustomerCount Lower Upper Outlier
State StatusDate
FL 2009-01-12 901 450.5 1351.5 False
2009-02-02 653 326.5 979.5 False
2009-03-23 752 376.0 1128.0 False
2009-04-06 1086 543.0 1629.0 False
2009-06-08 649 324.5 973.5 False

我们创建一个单独的 dataframe,叫 ALL,仅用 StatusDate 来为 Daily 数据集做索引。 我们简单地去掉 State 这一列。 Max 列则代表了每一个月最大的客户数量。 Max 列是用来是的图形更顺滑的。

  1. # 合并所有市场的
  2. # 按日期计算出最大的客户数
  3. ALL = pd.DataFrame(Daily['CustomerCount'].groupby(Daily.index.get_level_values(1)).sum())
  4. ALL.columns = ['CustomerCount'] # rename column
  5. # 按照年和月来分组
  6. YearMonth = ALL.groupby([lambda x: x.year, lambda x: x.month])
  7. # 找出每一个年和月的组合中最大的客户数
  8. ALL['Max'] = YearMonth['CustomerCount'].transform(lambda x: x.max())
  9. ALL.head()









































CustomerCount Max
StatusDate
2009-01-05 877 901
2009-01-12 901 901
2009-01-19 522 901
2009-02-02 953 953
2009-02-23 710 953

从上面的 ALL dataframe 中可以看到,在 2009年1月的这个月份,最大的客户数是 901。 如果我们使用 apply 的方式,我们将会得到一个以 (年 和 月)组合作为索引的 dataframe ,只有 Max 这一列有901这个值。


如果当前的客户数达到了公司制定的一定的目标值,这也会是一个很有趣的度量值。 现在的任务是可视化的展示当前的客户数是否达到了下面列出的目标值。 我们把这些目标叫做 BHAG (Big Hairy Annual Goal,年度战略目标)。

  • 12/31/2011 - 1,000 客户
  • 12/31/2012 - 2,000 客户
  • 12/31/2013 - 3,000 客户

我们将用 date_range 函数来创建日期。

定义: date_range(start=None, end=None, periods=None, freq=’D’, tz=None, normalize=False, name=None, closed=None)
文档(Docstring): 返回一个固定频率的日期时间索引,用日历天作为默认的频率

把频率设定为 A 或者是“年度” 我们将等到上述三个年份。

  1. pd.date_range?
  1. # 创建 BHAG 数据
  2. data = [1000, 2000, 3000]
  3. idx = pd.date_range(start='12/31/2011', end='12/31/2013', freq='A')
  4. BHAG = pd.DataFrame(data, index=idx, columns=['BHAG'])
  5. BHAG






















BHAG
2011-12-31 1000
2012-12-31 2000
2013-12-31 3000

我们之前学过用 concat 把 dataframe 合并。 记住,当我们设置 axis = 0 时我们按列合并 (row wise, 译者注: 即直接把列合并进去,行方向数据缺失用NaN来填充)。

  1. # 把 BHAG 和 ALL 两个数据集合并在一起
  2. combined = pd.concat([ALL,BHAG], axis=0)
  3. combined = combined.sort_index(axis=0)
  4. combined.tail()










































BHAG CustomerCount Max
2012-11-19 NaN 136.0 1115.0
2012-11-26 NaN 1115.0 1115.0
2012-12-10 NaN 1269.0 1269.0
2012-12-31 2000.0 NaN NaN
2013-12-31 3000.0 NaN NaN

  1. fig, axes = plt.subplots(figsize=(12, 7))
  2. combined['BHAG'].fillna(method='pad').plot(color='green', label='BHAG')
  3. combined['Max'].plot(color='blue', label='All Markets')
  4. plt.legend(loc='best');

png

这里还有一个需求是预测下一个年度的客户数,我们之后会通过几个简单的步骤完成。 我们想把已经合并的 dataframe 按照 Year 来分组,并且计算出年度的最大客户数。 这样每一行就是一个年度的数据。

  1. # Group by Year and then get the max value per year
  2. Year = combined.groupby(lambda x: x.year).max()
  3. Year










































BHAG CustomerCount Max
2009 NaN 2452.0 2452.0
2010 NaN 2065.0 2065.0
2011 1000.0 2711.0 2711.0
2012 2000.0 2061.0 2061.0
2013 3000.0 NaN NaN

  1. # 增加一列,表示为每一年比上一年变化的百分比
  2. Year['YR_PCT_Change'] = Year['Max'].pct_change(periods=1)
  3. Year
















































BHAG CustomerCount Max YR_PCT_Change
2009 NaN 2452.0 2452.0 NaN
2010 NaN 2065.0 2065.0 -0.157830
2011 1000.0 2711.0 2711.0 0.312833
2012 2000.0 2061.0 2061.0 -0.239764
2013 3000.0 NaN NaN NaN

要得到下一个年度末的客户数,我们假定当前的增长速率是维持恒定的。 我们按照这个增长速率来预测下一个年度的客户数量。

  1. (1 + Year.ix[2012,'YR_PCT_Change']) * Year.ix[2012,'Max']
  1. 1566.8465510881595

表示数据

为每一个州绘制单独的图表。

  1. # 第一张图是整个市场的
  2. ALL['Max'].plot(figsize=(10, 5));plt.title('ALL Markets')
  3. # 后面四张
  4. fig, axes = plt.subplots(nrows=2, ncols=2, figsize=(20, 10))
  5. fig.subplots_adjust(hspace=1.0) ## Create space between plots
  6. Daily.loc['FL']['CustomerCount']['2012':].fillna(method='pad').plot(ax=axes[0,0])
  7. Daily.loc['GA']['CustomerCount']['2012':].fillna(method='pad').plot(ax=axes[0,1])
  8. Daily.loc['TX']['CustomerCount']['2012':].fillna(method='pad').plot(ax=axes[1,0])
  9. Daily.loc['NY']['CustomerCount']['2012':].fillna(method='pad').plot(ax=axes[1,1])
  10. # 增加图表的抬头
  11. axes[0,0].set_title('Florida')
  12. axes[0,1].set_title('Georgia')
  13. axes[1,0].set_title('Texas')
  14. axes[1,1].set_title('North East');

png

png

This tutorial was created by HEDARO


本教程由派兰数据翻译

These tutorials are also available through an email course, please visit http://www.hedaro.com/pandas-tutorial to sign up today.