计算与推断思维 七、函数和表格

七、函数和表格

原文:Functions and Tables

译者:飞龙

协议:CC BY-NC-SA 4.0

自豪地采用谷歌翻译

通过使用 Python 中已有的函数,我们正在建立一个使用的技术清单,用于识别数据集中的规律和主题。 现在我们将探索Python编程语言的核心功能:函数定义。

我们在本书中已经广泛使用了函数,但从未定义过我们自己的函数。定义一个函数的目的是,给一个计算过程命名,它可能会使用多次。计算中有许多需要重复计算的情况。 例如,我们常常希望对表的列中的每个值执行相同的操作。

定义函数

double函数的定义仅仅使一个数值加倍。

# Our first function definition

def double(x):
    """ Double x """
    return 2*x

我们通过编写def来开始定义任何函数。 下面是这个小函数的其他部分(语法)的细分:

当我们运行上面的单元格时,没有使特定的数字加倍,并且double主体中的代码还没有求值。因此,我们的函数类似于一个菜谱。 每次我们遵循菜谱中的指导,我们都需要以食材开始。 每次我们想用我们的函数来使一个数字加倍时,我们需要指定一个数字。

我们可以用和调用其他函数完全相同的方式,来调用double。 每次我们这样做的时候,主体中的代码都会执行,参数的值赋给了名称x

double(17)
34
double(-0.6/4)
-0.3

以上两个表达式都是调用表达式。 在第二个里面,计算了表达式-0.6 / 4的值,然后将其作为参数x传递给double函数。 每个调用表达式最终都会执行double的主体,但使用不同的x值。

double的主体只有一行:

return 2*x

执行这个return语句会完成double函数体的执行,并计算调用表达式的值。

double的参数可以是任何表达式,只要它的值是一个数字。 例如,它可以是一个名称。 double函数不知道或不在意如何计算或存储参数。 它唯一的工作是,使用传递给它的参数的值来执行它自己的主体。

any_name = 42
double(any_name)
84

参数也可以是任何可以加倍的值。例如,可以将整个数值数组作为参数传递给double,结果将是另一个数组。

double(make_array(3, 4, 5))
array([ 6,  8, 10])

但是,函数内部定义的名称(包括像double的x这样的参数)只存在一小会儿。 它们只在函数被调用的时候被定义,并且只能在函数体内被访问。 我们不能在double之外引用x。 技术术语是x具有局部作用域。

因此,即使我们在上面的单元格中调用了double,名称x也不能在函数体外识别。

x
---------------------------------------------------------------------------
NameError                                 Traceback (most recent call last)
<ipython-input-18-401b30e3b8b5> in <module>()
----> 1 x

NameError: name 'x' is not defined

文档字符串。 虽然double比较容易理解,但是很多函数执行复杂的任务,并且没有解释就很难使用。 (你自己也可能已经发现了!)因此,一个组成良好的函数有一个唤起它的行为的名字,以及文档。 在 Python 中,这被称为文档字符串 - 描述了它的行为和对其参数的预期。 文档字符串也可以展示函数的示例调用,其中调用前面是>>>

文档字符串可以是任何字符串,只要它是函数体中的第一个东西。 文档字符串通常在开始和结束处使用三个引号来定义,这允许字符串跨越多行。 第一行通常是函数的完整但简短的描述,而下面的行则为将来的用户提供了进一步的指导。

下面是一个名为percent的函数定义,它带有两个参数。定义包括一个文档字符串。

# A function with more than one argument

def percent(x, total):
    """Convert x to a percentage of total.

    More precisely, this function divides x by total,
    multiplies the result by 100, and rounds the result
    to two decimal places.

    >>> percent(4, 16)
    25.0
    >>> percent(1, 6)
    16.67
    """
    return round((x/total)*100, 2)
percent(33, 200)
16.5

将上面定义的函数percent与下面定义的函数percents进行对比。 后者以数组为参数,将数组中的所有数字转换为数组中所有值的百分数。 百分数都四舍五入到两位,这次使用round来代替np.round,因为参数是一个数组而不是一个数字。

def percents(counts):
    """Convert the values in array_x to percents out of the total of array_x."""
    total = counts.sum()
    return np.round((counts/total)*100, 2)

函数percents返回一个百分数的数组,除了四舍五入之外,它总计是 100。

some_array = make_array(7, 10, 4)
percents(some_array)
array([ 33.33,  47.62,  19.05])

理解 Python 执行函数的步骤是有帮助的。 为了方便起见,我们在下面的同一个单元格中放入了函数定义和对这个函数的调用。

def biggest_difference(array_x):
    """Find the biggest difference in absolute value between two adjacent elements of array_x."""
    diffs = np.diff(array_x)
    absolute_diffs = abs(diffs)
    return max(absolute_diffs)

some_numbers = make_array(2, 4, 5, 6, 4, -1, 1)
big_diff = biggest_difference(some_numbers)
print("The biggest difference is", big_diff)
The biggest difference is 5

这就是当我们运行单元格时,所发生的事情。

多个参数

可以有多种方式来推广一个表达式或代码块,因此一个函数可以有多个参数,每个参数决定结果的不同方面。 例如,我们以前定义的百分比percents,每次都四舍五入到两位。 以下两个参的数定义允许不同调用四舍五入到不同的位数。

def percents(counts, decimal_places):
    """Convert the values in array_x to percents out of the total of array_x."""
    total = counts.sum()
    return np.round((counts/total)*100, decimal_places)

parts = make_array(2, 1, 4)
print("Rounded to 1 decimal place: ", percents(parts, 1))
print("Rounded to 2 decimal places:", percents(parts, 2))
print("Rounded to 3 decimal places:", percents(parts, 3))
Rounded to 1 decimal place:  [ 28.6  14.3  57.1]
Rounded to 2 decimal places: [ 28.57  14.29  57.14]
Rounded to 3 decimal places: [ 28.571  14.286  57.143]

这个新定义的灵活性来源于一个小的代价:每次调用该函数时,都必须指定小数位数。默认参数值允许使用可变数量的参数调用函数;在调用表达式中未指定的任何参数都被赋予其默认值,这在def语句的第一行中进行了说明。 例如,在percents的最终定义中,可选参数decimal_places赋为默认值2

def percents(counts, decimal_places=2):
    """Convert the values in array_x to percents out of the total of array_x."""
    total = counts.sum()
    return np.round((counts/total)*100, decimal_places)

parts = make_array(2, 1, 4)
print("Rounded to 1 decimal place:", percents(parts, 1))
print("Rounded to the default number of decimal places:", percents(parts))
Rounded to 1 decimal place: [ 28.6  14.3  57.1]
Rounded to the default number of decimal places: [ 28.57  14.29  57.14]

注:方法

函数通过将参数表达式放入函数名称后面的括号来调用。 任何独立定义的函数都是这样调用的。 你也看到了方法的例子,这些方法就像函数一样,但是用点符号来调用,比如some_table.sort(some_label)。 你定义的函数将始终首先使用函数名称,并传入所有参数来调用。

在列上应用函数

我们已经看到很多例子,通过将函数应用于现有列或其他数组,来创建新的表格的列。 所有这些函数都以数组作为参数。 但是我们经常打算,通过一个函数转换列中的条目,它不将数组作为它的函数。 例如,它可能只需要一个数字作为它的参数,就像下面定义的函数cut_off_at_100

def cut_off_at_100(x):
    """The smaller of x and 100"""
    return min(x, 100)
cut_off_at_100(17)
17
cut_off_at_100(117)
100
cut_off_at_100(100)
100

如果参数小于或等于 100,函数cut_off_at_100只返回它的参数。但是如果参数大于 100,则返回 100。

在我们之前使用人口普查数据的例子中,我们看到变量AGE的值为 100,表示“100 岁以上”。 以这种方式将年龄限制在 100 岁,正是cut_off_at_100所做的。

为了一次性对很多年龄使用这个函数,我们必须能够引用函数本身,而不用实际调用它。 类似地,我们可能会向厨师展示一个蛋糕的菜谱,并要求她用它来烤 6 个蛋糕。 在这种情况下,我们不会使用这个配方自己烘烤蛋糕, 我们的角色只是把菜谱给厨师。 同样,我们可以要求一个表格,在列中的 6 个不同的数字上调用cut_off_at_100

首先,我们创建了一个表,一列是人,一列是它们的年龄。 例如,C是 52 岁。

ages = Table().with_columns(
    'Person', make_array('A', 'B', 'C', 'D', 'E', 'F'),
    'Age', make_array(17, 117, 52, 100, 6, 101)
)
ages
PersonAge
A17
B117
C52
D100
E6
F101

应用

要在 100 岁截断年龄,我们将使用一个新的Table方法。 apply方法在列的每个元素上调用一个函数,形成一个返回值的新数组。 为了指出要调用的函数,只需将其命名(不带引号或括号)。 输入值的列的名称必须是字符串,仍然出现在引号内。

ages.apply(cut_off_at_100, 'Age')
array([ 17, 100,  52, 100,   6, 100])

我们在这里所做的是,将cut_off_at_100函数应用于age表的Age列中的每个值。 输出是函数的相应返回值的数组。 例如,17 还是 17,117 变成了 100,52 还是 52,等等。

此数组的长度与age表中原始Age列的长度相同,可用作名为Cut Off Age的新列中的值,并与现有的PersonAge列共存。

ages.with_column(
    'Cut Off Age', ages.apply(cut_off_at_100, 'Age')
)
PersonAgeCut Off Age
A1717
B117100
C5252
D100100
E66
F101100

作为值的函数

我们已经看到,Python 有很多种值。 例如,6是一个数值,"cake"是一个文本值,Table()是一个空表,age是一个表值(因为我们在上面定义)的名称。

在 Python 中,每个函数(包括cut_off_at_100)也是一个值。 这有助于再次考虑菜谱。 蛋糕的菜谱是一个真实的东西,不同于蛋糕或配料,你可以给它一个名字,像“阿尼的蛋糕菜谱”。 当我们用def语句定义cut_off_at_100时,我们实际上做了两件事情:我们创建了一个函数来截断数字 100,我们给它命名为cut_off_at_100

我们可以引用任何函数,通过写下它的名字,而没有实际调用它必需的括号或参数。当我们在上面调用apply时,我们做了这个。 当我们自己写下一个函数的名字,作为单元格中的最后一行时,Python 会生成一个函数的文本表示,就像打印一个数字或一个字符串值一样。

cut_off_at_100
<function __main__.cut_off_at_100>

请注意,我们没有使用引号(它只是一段文本)或cut_off_at_100()(它是一个函数调用,而且是无效的)。我们只是写下cut_off_at_100来引用这个函数。

就像我们可以为其他值定义新名称一样,我们可以为函数定义新名称。 例如,假设我们想把我们的函数称为cut_off,而不是cut_off_at_100。 我们可以这样写:

cut_off = cut_off_at_100

现在cut_off就是函数名称了。它是cut_off_at_100的相同函数。所以打印出的值应该相同。

cut_off
<function __main__.cut_off_at_100>

让我们看看另一个apply的应用。

示例:预测

数据科学经常用来预测未来。 如果我们试图预测特定个体的结果 - 例如,她将如何回应处理方式,或者他是否会购买产品,那么将预测基于其他类似个体的结果是很自然的。

查尔斯·达尔文(Charles Darwin)的堂兄弗朗西斯·高尔顿(Sir Francis Galton)是使用这个思想来基于数值数据进行预测的先驱。 他研究了物理特征是如何传递下来的。

下面的数据是父母和他们的成年子女的身高测量值,由高尔顿仔细收集。 每行对应一个成年子女。 变量是家庭的数字代码,父母的身高(以英寸为单位),“双亲身高”,这是父母双方身高的加权平均值 [1],家庭中子女的数量 ,以及子女的出生次序(第几个),性别和身高。

[1] 高尔顿在计算男性和女性的平均身高之前,将女性身高乘上 1.08。对于这个的讨论,请查看 Chance,这是一个由美国统计协会出版的杂志。

# Galton's data on heights of parents and their adult children
galton = Table.read_table('galton.csv')
galton
familyfathermothermidparentHeightchildrenchildNumgenderchildHeight
178.56775.4341male73.2
178.56775.4342female69.2
178.56775.4343female69
178.56775.4344female69
275.566.573.6641male73.5
275.566.573.6642male72.5
275.566.573.6643female65.5
275.566.573.6644female65.5
3756472.0621male71
3756472.0622female68

(省略了 924 行)

收集数据的主要原因是,能够预测父母所生的子女的成年身高,其中父母和数据集中的类似。让我们尝试这样做,用双亲的身高作为我们预测的基础变量。 因此双亲的身高是我们的预测性变量。

表格heights包含双亲和子女的身高。 两个变量的散点图显示了正相关,正如我们对这些变量的预期。

heights = galton.select(3, 7).relabeled(0, 'MidParent').relabeled(1, 'Child')
heights
MidParentChild
75.4373.2
75.4369.2
75.4369
75.4369
73.6673.5
73.6672.5
73.6665.5
73.6665.5
72.0671
72.0668

(省略了 924 行)

heights.scatter(0)

现在假设高尔顿遇到了新的一对夫妇,与他的数据集类似,并且想知道他们的子女有多高。考虑到双亲身高是 68 英寸,他预测子女身高的一个好方法是什么?

一个合理的方法是基于约 68 英寸的双亲身高对应的所有点,来做预测。预测值等于从这些点计算的子女身高的均值。

假设我们是高尔顿,并执行这个计划。现在我们只是对“68 英寸左右”的含义做一个合理的定义,并用它来处理。在课程的后面,我们将研究这种选择的后果。

我们的“接近”的意思是“在半英寸之内”。下图显示了 67.5 英寸和 68.5 英寸之间的双亲身高对应的所有点。这些都是红色直线之间的点。每一个点都对应一个子女;我们对新夫妇的子女身高的预测是所有子女的平均身高。这由金色的点表示。

忽略代码,仅仅专注于理解到达金色的点的心理过程。

heights.scatter('MidParent')
_ = plots.plot([67.5, 67.5], [50, 85], color='red', lw=2)
_ = plots.plot([68.5, 68.5], [50, 85], color='red', lw=2)
_ = plots.scatter(68, 66.24, color='gold', s=40)

为了准确计算出金色的点的位置,我们首先需要确定直线之间的所有点。 这些点对应于MidParent在 67.5 英寸和 68.5 英寸之间的行。

close_to_68 = heights.where('MidParent', are.between(67.5, 68.5))
close_to_68
MidParentChild
68.4462
67.9471.2
67.9467
68.3362.5
68.2373
68.2372
68.2369
67.9873
67.9871
67.9871

(省略了 121 行)

双亲身高为 68 英寸的子女的预测身高,是这些行中子女的平均身高。 这是 66.24 英寸。

close_to_68.column('Child').mean()
66.24045801526718

我们现在有了一种方法,给定任何数据集中的双亲身高,就可以预测子女的身高。我们可以定义一个函数predict_child来实现它。 除了名称的选择之外,函数的主体由上面两个单元格中的代码组成。

def predict_child(mpht):
    """Predict the height of a child whose parents have a midparent height of mpht.

    The prediction is the average height of the children whose midparent height is
    in the range mpht plus or minus 0.5.
    """

    close_points = heights.where('MidParent', are.between(mpht-0.5, mpht + 0.5))
    return close_points.column('Child').mean() 

给定 68 英寸的双亲身高,函数predict_child返回与之前相同的预测(66.24 英寸)。 定义函数的好处在于,我们可以很容易地改变预测变量的值,并得到一个新的预测结果。

predict_child(68)
66.24045801526718
predict_child(74)
70.415789473684214

这些预测有多好? 我们可以了解它,通过将预测值与我们已有的数据进行比较。 为此,我们首先将函数predict_child应用于Midparent列,并将结果收入称为Prediction的新列中。

# Apply predict_child to all the midparent heights

heights_with_predictions = heights.with_column(
    'Prediction', heights.apply(predict_child, 'MidParent')
)
heights_with_predictions
MidParentChildPrediction
75.4373.270.1
75.4369.270.1
75.436970.1
75.436970.1
73.6673.570.4158
73.6672.570.4158
73.6665.570.4158
73.6665.570.4158
72.067168.5025
72.066868.5025

(省略了 924 行)

为了查看预测值相对于观察数据的位置,可以使用MidParent作为公共水平轴绘制重叠的散点图。

heights_with_predictions.scatter('MidParent')

金色的点的图形称为均值图,因为每个金色的点都是两条直线的中心,就像之前绘制的那样。每个都按照给定的双亲高度,做出了子女高度的预测。例如,散点图显示,对于 72 英寸的双亲高度,子女的预测高度将在 68 英寸和 69 英寸之间,事实上,predict_child(72)返回 68.5。

高尔顿的计算和可视化与我们非常相似,除了他没有 Python。他通过散点图绘制了均值图,并注意到它大致沿着直线。这条直线现在被称为回归线,是最常见的预测方法之一。高尔顿的朋友,数学家卡尔·皮尔森(Karl Pearson)用这些分析来形式化关联的概念。

这个例子,就像约翰·斯诺(John Snow)对霍乱死亡的分析一样,说明了现代数据科学的一些基本概念的根源可追溯到一个多世纪之前。高尔顿的方法,比如我们在这里使用的方法,是最近邻预测方法的雏形,现在在不同的环境中有着有效的应用。机器学习的现代领域包括这些方法的自动化,来基于庞大且快速发展的数据集进行预测。

按照单变量分类

数据科学家经常需要根据共有的特征,将个体分成不同的组,然后确定组的一些特征。 例如,在使用高尔顿高度数据的例子中,我们看到根据父母的平均高度对家庭进行分类,然后找出每个小组中子女的平均身高,较为实用。

这部分关于将个体分类到非数值类别。我们从回顾gourp的基本用法开始。

计算每个分类的数量

具有单个参数的group方法计算列中每个值的数量。 结果中,分组列(用于分组的列)中的每个唯一值是一行。

这是一个关于冰淇淋圆通的小型数据表。 group方法可以用来列出不同的口味,并提供每种口味的计数。

cones = Table().with_columns(
    'Flavor', make_array('strawberry', 'chocolate', 'chocolate', 'strawberry', 'chocolate'),
    'Price', make_array(3.55, 4.75, 6.55, 5.25, 5.25)
)
cones
FlavorPrice
strawberry3.55
chocolate4.75
chocolate6.55
strawberry5.25
chocolate5.25
cones.group('Flavor')
Flavorcount
chocolate3
strawberry2

有两个不同的类别,巧克力和草莓。 group的调用会在每个类别中创建一个计数表。 该列默认称为count,并包含每个类别中的行数。

注意,这一切都可以从Flavor列中找到。Price列尚未使用。

但是如果我们想要每种不同风味的圆筒的总价格呢? 这是group的第二个参数的作用。

发现每个类别的特征

group的可选的第二个参数是一个函数,用于聚合所有这些行的其他列中的值。 例如,sum将累计与每个类别匹配的所有行中的价格。 这个结果中,分组列中每个唯一值是一行,但与原始表列数相同。

为了找到每种口味的总价格,我们再次调用group,用Flavor作为第一个参数。 但这一次有第二个参数:函数名称sum

cones.group('Flavor', sum)
FlavorPrice sum
chocolate16.55
strawberry8.8

为了创建这个新表格,group已经计算了对应于每种不同口味的,所有行中的Price条目的总和。 三个chocolate行的价格共计$16.55(你可以假设价格是以美元计量的)。 两个strawberry行的价格共计8.80

新创建的“总和”列的标签是Price sum,它通过使用被求和列的标签,并且附加单词sum创建。

由于group计算除了类别之外的所有列的sum,因此不需要指定必须对价格求和。

为了更详细地了解group在做什么,请注意,你可以自己计算总价格,不仅可以通过心算,还可以使用代码。 例如,要查找所有巧克力圆筒的总价格,你可以开始创建一个仅包含巧克力圆筒的新表,然后访问价格列:

cones.where('Flavor', are.equal_to('chocolate')).column('Price')
array([ 4.75,  6.55,  5.25])
sum(cones.where('Flavor', are.equal_to('chocolate')).column('Price'))
16.550000000000001

这就是groupFlavor中每个不同的值所做的事情。

# For each distinct value in `Flavor, access all the rows
# and create an array of `Price`

cones_choc = cones.where('Flavor', are.equal_to('chocolate')).column('Price')
cones_strawb = cones.where('Flavor', are.equal_to('strawberry')).column('Price')

# Display the arrays in a table

grouped_cones = Table().with_columns(
    'Flavor', make_array('chocolate', 'strawberry'),
    'Array of All the Prices', make_array(cones_choc, cones_strawb)
)

# Append a column with the sum of the `Price` values in each array

price_totals = grouped_cones.with_column(
    'Sum of the Array', make_array(sum(cones_choc), sum(cones_strawb))
)
price_totals
FlavorArray of All the PricesSum of the Array
chocolate[ 4.75 6.55 5.25]16.55
strawberry[ 3.55 5.25]8.8

你可以用任何其他可以用于数组的函数来替换sum。 例如,你可以使用max来查找每个类别中的最大价格:

cones.group('Flavor', max)
FlavorPrice max
chocolate6.55
strawberry5.25

同样,group在每个Flavor分类中创建价格数组,但现在它寻找每个数组的max

price_maxes = grouped_cones.with_column(
    'Max of the Array', make_array(max(cones_choc), max(cones_strawb))
)
price_maxes
FlavorArray of All the PricesMax of the Array
chocolate[ 4.75 6.55 5.25]6.55
strawberry[ 3.55 5.25]5.25

实际上,只有一个参数的原始调用,与使用len作为函数并清理表格的效果相同。

lengths = grouped_cones.with_column(
    'Length of the Array', make_array(len(cones_choc), len(cones_strawb))
)
lengths
FlavorArray of All the PricesLength of the Array
chocolate[ 4.75 6.55 5.25]3
strawberry[ 3.55 5.25]2

示例:NBA 薪水

nba表包含了 2015~2016 年 NBA 球员的数据。 我们早些时候审查了这些数据。 回想一下,薪水以百万美元计算。

nba1 = Table.read_table('nba_salaries.csv')
nba = nba1.relabeled("'15-'16 SALARY", 'SALARY')
nba
PLAYERPOSITIONTEAMSALARY
Paul MillsapPFAtlanta Hawks18.6717
Al HorfordCAtlanta Hawks12
Tiago SplitterCAtlanta Hawks9.75625
Jeff TeaguePGAtlanta Hawks8
Kyle KorverSGAtlanta Hawks5.74648
Thabo SefoloshaSFAtlanta Hawks4
Mike ScottPFAtlanta Hawks3.33333
Kent BazemoreSFAtlanta Hawks2
Dennis SchroderPGAtlanta Hawks1.7634
Tim Hardaway Jr.SGAtlanta Hawks1.30452

(省略了 407 行)

(1)每支球队为球员的工资支付了多少钱?

唯一涉及的列是TEAMSALARY。 我们必须按TEAM对这些行进行分组,然后对这些分类的工资进行求和。

teams_and_money = nba.select('TEAM', 'SALARY')
teams_and_money.group('TEAM', sum)
TEAMSALARY sum
Atlanta Hawks69.5731
Boston Celtics50.2855
Brooklyn Nets57.307
Charlotte Hornets84.1024
Chicago Bulls78.8209
Cleveland Cavaliers102.312
Dallas Mavericks65.7626
Denver Nuggets62.4294
Detroit Pistons42.2118
Golden State Warriors94.0851

(省略了 20 行)

(2)五个位置的每个中有多少个 NBA 球员呢?

我们必须按POSITION分类并计数。 这可以通过一个参数来完成:

nba.group('POSITION')
POSITIONcount
C69
PF85
PG85
SF82
SG96

(3)五个位置的每个中,球员平均薪水是多少?

这一次,我们必须按POSITION分组,并计算薪水的均值。 为了清楚起见,我们将用一张表格来描述位置和薪水。

positions_and_money = nba.select('POSITION', 'SALARY')
positions_and_money.group('POSITION', np.mean)
POSITIONSALARY mean
C6.08291
PF4.95134
PG5.16549
SF5.53267
SG3.9882

中锋是最高薪的职位,均值超过 600 万美元。

如果我们开始没有选择这两列,那么group不会尝试对nba中的类别列计算“平均”。 (“亚特兰大老鹰”和“波士顿凯尔特人队”这两个字符串是不可能平均)。它只对数值列做算术,其余的都是空白的。

nba.group('POSITION', np.mean)
POSITIONPLAYER meanTEAM meanSALARY mean
C6.08291
PF4.95134
PG5.16549
SF5.53267
SG3.9882

交叉分类

通过多个变量的交叉分类

当个体具有多个特征时,有很多不同的对他们分类的方法。 例如,如果我们有大学生的人口数据,对于每个人我们都有专业和大学的年数,那么这些学生就可以按照专业,按年份,或者是专业和年份的组合来分类。

group方法也允许我们根据多个变量划分个体。 这被称为交叉分类。

两个变量:计算每个类别偶对的数量

more_cones表记录了六个冰淇淋圆筒的味道,颜色和价格。

more_cones = Table().with_columns(
    'Flavor', make_array('strawberry', 'chocolate', 'chocolate', 'strawberry', 'chocolate', 'bubblegum'),
    'Color', make_array('pink', 'light brown', 'dark brown', 'pink', 'dark brown', 'pink'),
    'Price', make_array(3.55, 4.75, 5.25, 5.25, 5.25, 4.75)
)

more_cones
FlavorColorPrice
strawberrypink3.55
chocolatelight brown4.75
chocolatedark brown5.25
strawberrypink5.25
chocolatedark brown5.25
bubblegumpink4.75

我们知道如何使用group,来计算每种口味的冰激凌圆筒的数量。

more_cones.group('Flavor')
Flavorcount
bubblegum1
chocolate3
strawberry2

但是现在每个圆筒也有一个颜色。 为了将圆筒按风味和颜色进行分类,我们将把标签列表作为参数传递给group。 在分组列中出现的每个唯一值的组合,在生成的表格中都占一行。 和以前一样,一个参数(这里是一个列表,但是也可以是一个数组)提供了行数。

虽然有六个圆筒,但只有四种风味和颜色的唯一组合。 两个圆筒是深褐色的巧克力,还有两个粉红色的草莓。

more_cones.group(['Flavor', 'Color'])
FlavorColorcount
bubblegumpink1
chocolatedark brown2
chocolatelight brown1
strawberrypink2

两个变量:查找每个类别偶对的特征

第二个参数聚合所有其他列,它们不在分组列的列表中。

more_cones.group(['Flavor', 'Color'], sum)
FlavorColorPrice sum
bubblegumpink4.75
chocolatedark brown10.5
chocolatelight brown4.75
strawberrypink8.8

三个或更多的变量。 你可以使用group,按三个或更多类别变量对行分类。 只要将它们全部包含列表中,它是第一个参数。 但是由多个变量交叉分类可能会变得复杂,因为不同类别组合的数量可能相当大。

数据透视表:重新排列group的输出

交叉分类的许多使用只涉及两个类别变量,如上例中的FlavorColor。 在这些情况下,可以在不同类型的表中显示分类结果,称为数据透视表(pivot table)。 数据透视表,也被称为列联表(contingency table),可以更容易地处理根据两个变量进行分类的数据。

回想一下,使用group来计算每个风味和颜色的类别偶对的圆筒数量:

more_cones.group(['Flavor', 'Color'])
FlavorColorcount
bubblegumpink1
chocolatedark brown2
chocolatelight brown1
strawberrypink2

使用Tablepivot方法可以以不同方式展示相同数据。暂时忽略这些代码,然后查看所得表。

more_cones.pivot('Flavor', 'Color')
Colorbubblegumchocolatestrawberry
dark brown020
light brown010
pink102

请注意,此表格显示了所有九种可能的风味和颜色偶对,包括我们的数据中不存在的偶对,比如“深棕色泡泡糖”。 还要注意,每个偶对中的计数都出现在表格的正文中:要找到浅棕色巧克力圆筒的数量,用眼睛沿着浅棕色的行看,直到它碰到巧克力一列。

group方法接受两个标签的列表,因为它是灵活的:可能需要一个或三个或更多。 另一方面,数据透视图总是需要两个列标签,一个确定列,一个确定行。

pivot方法与group方法密切相关:group将拥有相同值的组合的行分组在一起。它与group不同,因为它将所得值组织在一个网格中。 pivot的第一个参数是列标签,包含的值将用于在结果中形成新的列。第二个参数是用于行的列标签。结果提供了原始表的所有行的计数,它们拥有相同的行和列值组合。

group一样,pivot可以和其他参数一同使用,来发现每个类别组合的特征。名为values的第三个可选参数表示一列值,它们替换网格的每个单元格中的计数。所有这些值将不会显示,但是;第四个参数collect表示如何将它们全部汇总到一个聚合值中,来显示在单元格中。

用例子来澄清这一点。这里是一个透视表,用于寻找每个单元格中的圆筒的总价格。

more_cones.pivot('Flavor', 'Color', values='Price', collect=sum)
Colorbubblegumchocolatestrawberry
dark brown010.50
light brown04.750
pink4.7508.8

这里group做了同一件事。

more_cones.group(['Flavor', 'Color'], sum)
FlavorColorPrice sum
bubblegumpink4.75
chocolatedark brown10.5
chocolatelight brown4.75
strawberrypink8.8

尽管两个表中的数字都相同,但由pivot生成的表格更易于阅读,因而更易于分析。 透视表的优点是它将分组的值放到相邻的列中,以便它们可以进行组合和比较。

示例:加州成人的教育和收入

加州的开放数据门户是丰富的加州生活的信息来源。 这是 2008 至 2014 年间加利福尼亚州教育程度和个人收入的数据集。数据来源于美国人口普查的当前人口调查。

对于每年,表格都记录了加州的Population Count(人口数量),按照年龄,性别,教育程度和个人收入,构成不同的组合。 我们将只研究 2014 年的数据。

full_table = Table.read_table('educ_inc.csv')
ca_2014 = full_table.where('Year', are.equal_to('1/1/14 0:00')).where('Age', are.not_equal_to('00 to 17'))
ca_2014
YearAgeGenderEducational AttainmentPersonal IncomePopulation Count
1/1/14 0:0018 to 64FemaleNo high school diplomaH: 75,000 and over2058
1/1/14 0:0065 to 80+MaleNo high school diplomaH: 75,000 and over2153
1/1/14 0:0065 to 80+FemaleNo high school diplomaG: 50,000 to 74,9994666
1/1/14 0:0065 to 80+FemaleHigh school or equivalentH: 75,000 and over7122
1/1/14 0:0065 to 80+FemaleNo high school diplomaF: 35,000 to 49,9997261
1/1/14 0:0065 to 80+MaleNo high school diplomaG: 50,000 to 74,9998569
1/1/14 0:0018 to 64FemaleNo high school diplomaG: 50,000 to 74,99914635
1/1/14 0:0065 to 80+MaleNo high school diplomaF: 35,000 to 49,99915212
1/1/14 0:0065 to 80+MaleCollege, less than 4-yr degreeB: 5,000 to 9,99915423
1/1/14 0:0065 to 80+FemaleBachelor’s degree or higherA: 0 to 4,99915459

(省略了 117 行)

表中的每一行对应一组年龄,性别,教育程度和收入。 总共有 127 个这样的组合!

作为第一步,从一个或两个变量开始是个好主意。 我们只关注一对:教育程度和个人收入。

educ_inc = ca_2014.select('Educational Attainment', 'Personal Income', 'Population Count')
educ_inc
Educational AttainmentPersonal IncomePopulation Count
No high school diplomaH: 75,000 and over2058
No high school diplomaH: 75,000 and over2153
No high school diplomaG: 50,000 to 74,9994666
High school or equivalentH: 75,000 and over7122
No high school diplomaF: 35,000 to 49,9997261
No high school diplomaG: 50,000 to 74,9998569
No high school diplomaG: 50,000 to 74,99914635
No high school diplomaF: 35,000 to 49,99915212
College, less than 4-yr degreeB: 5,000 to 9,99915423
Bachelor’s degree or higherA: 0 to 4,99915459

(省略了 117 行)

我们先看看教育程度。 这个变量的分类已经由不同的收入水平细分了。 因此,我们将按照教育程度分组,并将每个分类中的人口数量相加。

education = educ_inc.select('Educational Attainment', 'Population Count')
educ_totals = education.group('Educational Attainment', sum)
educ_totals
Educational AttainmentPopulation Count sum
Bachelor’s degree or higher8525698
College, less than 4-yr degree7775497
High school or equivalent6294141
No high school diploma4258277

教育程度只有四类。 计数太大了,查看百分比更有帮助。 为此,我们将使用前面章节中定义的函数percents。 它将数值数组转换为输入数组总量的百分比数组。

def percents(array_x):
    return np.round( (array_x/sum(array_x))*100, 2)

我们现在有加州成人的教育程度分布。 超过 30% 的人拥有学士或更高学位,而几乎 16% 没有高中文凭。

educ_distribution = educ_totals.with_column(
    'Population Percent', percents(educ_totals.column(1))
)
educ_distribution
Educational AttainmentPopulation Count sumPopulation Percent
Bachelor’s degree or higher852569831.75
College, less than 4-yr degree777549728.96
High school or equivalent629414123.44
No high school diploma425827715.86

通过使用pivot,我们可以得到一张加州成人的透视表(计数表),按照Educational AttainmentPersonal Income交叉分类。

totals = educ_inc.pivot('Educational Attainment', 'Personal Income', values='Population Count', collect=sum)
totals
Personal IncomeBachelor’s degree or higherCollege, less than 4-yr degreeHigh school or equivalentNo high school diploma
A: 0 to 4,99957549198501111618731204529
B: 5,000 to 9,999326020810641626499597039
C: 10,000 to 14,999452449798596692661664607
D: 15,000 to 24,99977368413452571252377875498
E: 25,000 to 34,9996938841091642929218464564
F: 35,000 to 49,99911227911112421782804260579
G: 50,000 to 74,9991594681883826525517132516
H: 75,000 and over298669874810332319258945

在这里你可以看到pivot相比其他方法的威力。 计数的每一列都是个人收入在特定教育程度中的分布。 将计数转换为百分数可以让我们比较四个分布。

distributions = totals.select(0).with_columns(
    "Bachelor's degree or higher", percents(totals.column(1)),
    'College, less than 4-yr degree', percents(totals.column(2)),
    'High school or equivalent', percents(totals.column(3)),
    'No high school diploma', percents(totals.column(4))   
    )

distributions
Personal IncomeBachelor’s degree or higherCollege, less than 4-yr degreeHigh school or equivalentNo high school diploma
A: 0 to 4,9996.7512.6718.4628.29
B: 5,000 to 9,9993.8210.439.9514.02
C: 10,000 to 14,9995.3110.271115.61
D: 15,000 to 24,9999.0717.319.920.56
E: 25,000 to 34,9998.1414.0414.7610.91
F: 35,000 to 49,99913.1714.3112.446.12
G: 50,000 to 74,99918.711.378.353.11
H: 75,000 and over35.039.625.131.38

一眼就能看出,超过 35% 的学士或以上学位的收入达到 $75,000 美元以上,而其他教育分类中,少于 10% 的人达到了这一水平。

下面的条形图比较了没有高中文凭的加州成年人的个人收入分布情况,和完成学士或更高学位的人的收入分布情况。 分布的差异是惊人的。 教育程度与个人收入有明显的正相关关系。

distributions.select(0, 1, 4).barh(0)

按列连接表

通常,同一个人的数据在多个表格中维护。 例如,大学的一个办公室可能有每个学生完成学位的时间的数据,而另一个办公室则有学生学费和经济援助的数据。

为了了解学生的经历,将两个数据集放在一起可能会有帮助。 如果数据是在两个表中,每个学生都有一行,那么我们希望将这些列放在一起,确保行是匹配的,以便将每个学生的信息保持在一行上。

让我们在一个简单的示例的背景下实现它,然后在更大的数据集上使用这个方法。

圆筒表是我们以前遇到的。 现在假设每种口味的冰激凌的评分都在独立的表格中。

cones = Table().with_columns(
    'Flavor', make_array('strawberry', 'vanilla', 'chocolate', 'strawberry', 'chocolate'),
    'Price', make_array(3.55, 4.75, 6.55, 5.25, 5.75)
)
cones
FlavorPrice
strawberry3.55
vanilla4.75
chocolate6.55
strawberry5.25
chocolate5.75
ratings = Table().with_columns(
    'Kind', make_array('strawberry', 'chocolate', 'vanilla'),
    'Stars', make_array(2.5, 3.5, 4)
)
ratings
KindStars
strawberry2.5
chocolate3.5
vanilla4

每个表都有一个包含冰淇淋风味的列:conesFlavor列,ratingsKind列。 这些列中的条目可以用来连接两个表。

join方法创建一个新的表,其中cones表中的每个圆筒都增加了评分信息。 对于cones中的每个圆筒,join会找到ratings中的行,它的Kind匹配圆筒的Flavor。 我们必须告诉join使用这些列进行匹配。

rated = cones.join('Flavor', ratings, 'Kind')
rated
FlavorPriceStars
chocolate6.553.5
chocolate5.753.5
strawberry3.552.5
strawberry5.252.5
vanilla4.754

现在每个圆筒不仅拥有价格,而且还有风味评分。

一般来说,使用来自另一个表(比如table2)的信息来扩充一个表(如table1)的join调用如下所示:

table1.join(table1_column_for_joining, table2, table2_column_for_joining)

新的rated表使我们能够计算出价格与星星的比值,你可以把它看作是非正式的价值衡量标准。 低的值更好 - 它们意味着你为评分的每个星星花费更少。

rated.with_column('$/Star', rated.column('Price') / rated.column('Stars')).sort(3)
FlavorPriceStars$/Star
vanilla4.7541.1875
strawberry3.552.51.42
chocolate5.753.51.64286
chocolate6.553.51.87143
strawberry5.252.52.1

虽然草莓在这三种口味中评分最低,但是这个标准下草莓更加便宜,因为每颗星星的花费并不高。

警告。顺序很重要。由于join中的第二个表用于扩充第一个表,所以重要的是,第一个表中的每一行在第二个表中只有一个匹配的行。如果第一个表中的某一行在第二个表中没有匹配项,则信息可能丢失。如果第一个表中的某一行在第二个表中有多个匹配项,那么join将只选择一个,这也是一种信息丢失。

我们可以在下面的例子中看到它,它试图通过相同的两列连接相同的两个表格,但是以另一种顺序。这种连接是没有意义的:它试图用价格来扩展每种风味的评分,但是根据cones表,每种风味都有一个以上的圆筒(和价格)。结果是两个圆筒消失了。join方法仅仅在cones寻找对应chocolate的第一行,而忽略其他行。

ratings.join('Kind', cones, 'Flavor')
KindStarsPrice
chocolate3.56.55
strawberry2.53.55
vanilla44.75

假设有个冰淇淋的评分表,我们已经求出了每种风味的平均评分。

reviews = Table().with_columns(
    'Flavor', make_array('vanilla', 'chocolate', 'vanilla', 'chocolate'),
    'Stars', make_array(5, 3, 5, 4)
)
reviews
FlavorStars
vanilla5
chocolate3
vanilla5
chocolate4
average_review = reviews.group('Flavor', np.average)
average_review
FlavorStars average
chocolate3.5
vanilla5

我们可以连接conesaverage_review,通过提供用于连接的列标签。

cones.join('Flavor', average_review, 'Flavor')
FlavorPriceStars average
chocolate6.553.5
chocolate5.753.5
vanilla4.755

注意草莓圆筒是如何消失的。 没有草莓圆筒的评价,所以没有草莓的行可以连接的东西。 这可能是一个问题,也可能不是 - 这取决于我们试图使用连接表执行的分析。

湾区共享单车

在本章结尾,我们通过使用我们学过的所有方法,来检验新的大型数据集。 我们还将介绍一个强大的可视化工具map_table

湾区自行车共享服务公司在其系统中发布了一个数据集,描述了 2014 年 9 月到 2015 年 8 月期间的每个自行车的租赁。 总共有 354152 次出租。 表的列是:

  • 租赁 ID
  • 租赁的时间,以秒为单位
  • 开始日期
  • 起点站的名称和起始终端的代码
  • 终点站的名称和终止终端的代码
  • 自行车的序列号
  • 订阅者类型和邮政编码
trips = Table.read_table('trip.csv')
trips
Trip IDDurationStart DateStart StationStart TerminalEnd DateEnd StationEnd TerminalBike #Subscriber TypeZip Code
9134607658/31/2015 23:26Harry Bridges Plaza (Ferry Building)508/31/2015 23:39San Francisco Caltrain (Townsend at 4th)70288Subscriber2139
91345910368/31/2015 23:11San Antonio Shopping Center318/31/2015 23:28Mountain View City Hall2735Subscriber95032
9134553078/31/2015 23:13Post at Kearny478/31/2015 23:182nd at South Park64468Subscriber94107
9134544098/31/2015 23:10San Jose City Hall108/31/2015 23:17San Salvador at 1st868Subscriber95113
9134537898/31/2015 23:09Embarcadero at Folsom518/31/2015 23:22Embarcadero at Sansome60487Customer9069
9134522938/31/2015 23:07Yerba Buena Center of the Arts (3rd @ Howard)688/31/2015 23:12San Francisco Caltrain (Townsend at 4th)70538Subscriber94118
9134518968/31/2015 23:07Embarcadero at Folsom518/31/2015 23:22Embarcadero at Sansome60363Customer92562
9134502558/31/2015 22:16Embarcadero at Sansome608/31/2015 22:20Steuart at Market74470Subscriber94111
9134491268/31/2015 22:12Beale at Market568/31/2015 22:15Temporary Transbay Terminal (Howard at Beale)55439Subscriber94130
9134489328/31/2015 21:57Post at Kearny478/31/2015 22:12South Van Ness at Market66472Subscriber94702

(省略了 354142 行)

我们只专注于免费行程,这是持续不到 1800 秒(半小时)的行程。 长途行程需要付费。

下面的直方图显示,大部分行程需要大约 10 分钟(600 秒)左右。 很少有人花了近 30 分钟(1800 秒),可能是因为人们试图在截止时间之前退还自行车,以免付费。

commute = trips.where('Duration', are.below(1800))
commute.hist('Duration', unit='Second')

我们可以通过指定更多的桶来获得更多的细节。 但整体形状并没有太大变化。

commute.hist('Duration', bins=60, unit='Second')

使用grouppivot探索数据

我们可以使用group来识别最常用的起点站。

starts = commute.group('Start Station').sort('count', descending=True)
starts
Start Stationcount
San Francisco Caltrain (Townsend at 4th)25858
San Francisco Caltrain 2 (330 Townsend)21523
Harry Bridges Plaza (Ferry Building)15543
Temporary Transbay Terminal (Howard at Beale)14298
2nd at Townsend13674
Townsend at 7th13579
Steuart at Market13215
Embarcadero at Sansome12842
Market at 10th11523
Market at Sansome11023

(省略了 60 行)

大多数行程起始于 Townsend 的 Caltrain 站,和旧金山的四号车站。 人们乘坐火车进入城市,然后使用共享单车到达下一个目的地。

group方法也可以用于按照起点站和终点站,对租赁进行分类。

commute.group(['Start Station', 'End Station'])
Start StationEnd Stationcount
2nd at Folsom2nd at Folsom54
2nd at Folsom2nd at South Park295
2nd at Folsom2nd at Townsend437
2nd at Folsom5th at Howard113
2nd at FolsomBeale at Market127
2nd at FolsomBroadway St at Battery St67
2nd at FolsomCivic Center BART (7th at Market)47
2nd at FolsomClay at Battery240
2nd at FolsomCommercial at Montgomery128
2nd at FolsomDavis at Jackson28

(省略了 1619 行)

共有五十四次行程开始和结束于在 Folsom 二号车站,。 有很多人(437 人)往返于 Folsom 二号和 Townsend 二号车站之间。

pivot方法执行相同的分类,但将结果显示在一个透视表中,该表显示了起点和终点站的所有可能组合,即使其中一些不对应任何行程。 请记住,pivot函数的第一个参数指定了数据透视表的列标签;第二个参数指定行标签。

在 Beale at Market 附近有一个火车站以及一个湾区快速公交(BART)站,解释了从那里开始和结束的大量行程。

commute.pivot('Start Station', 'End Station')
End Station2nd at Folsom2nd at South Park2nd at Townsend5th at HowardAdobe on AlmadenArena Green / SAP CenterBeale at MarketBroadway St at Battery StCalifornia Ave Caltrain StationCastro Street and El Camino RealCivic Center BART (7th at Market)Clay at BatteryCommercial at MontgomeryCowper at UniversityDavis at JacksonEmbarcadero at BryantEmbarcadero at FolsomEmbarcadero at SansomeEmbarcadero at VallejoEvelyn Park and RideFranklin at MapleGolden Gate at PolkGrant Avenue at Columbus AvenueHarry Bridges Plaza (Ferry Building)Howard at 2ndJapantownMLK LibraryMarket at 10thMarket at 4thMarket at SansomeMechanics Plaza (Market at Battery)Mezes ParkMountain View Caltrain StationMountain View City HallPalo Alto Caltrain StationPark at OlivePaseo de San AntonioPost at KearnyPowell Street BARTPowell at Post (Union Square)Redwood City Caltrain StationRedwood City Medical CenterRedwood City Public LibraryRengstorff Avenue / California StreetRyland ParkSJSU - San Salvador at 9thSJSU 4th at San CarlosSan Antonio Caltrain StationSan Antonio Shopping CenterSan Francisco Caltrain (Townsend at 4th)San Francisco Caltrain 2 (330 Townsend)San Francisco City HallSan Jose City HallSan Jose Civic CenterSan Jose Diridon Caltrain StationSan Mateo County CenterSan Pedro SquareSan Salvador at 1stSanta Clara County Civic CenterSanta Clara at AlmadenSouth Van Ness at MarketSpear at FolsomSt James ParkStanford in Redwood CitySteuart at MarketTemporary Transbay Terminal (Howard at Beale)Townsend at 7thUniversity and EmersonWashington at KearnyYerba Buena Center of the Arts (3rd @ Howard)
2nd at Folsom54190554107004021004478540977324114001130416530016911430233000000601218800000000069444521000000003857003923734201731
2nd at South Park295164711800020885001128716003756178831160057735745000013919916331190000002998411300000000055948048000000006615200374429143063209
2nd at Townsend4371511859200608350008032916803863616585062540027315260729500110225845177000000120100141000000000905299140000000072508002349784417057166
5th at Howard11317714883005913000203761290305749166540085783714780030315816890000000931831690000000006901859480000000011610200182750200043267
Adobe on Almaden000011400000000000000000001770000000002500000007716000001923265020451000140000000
Arena Green / SAP Center000076400000000000000000001650000000002100000002437000006207056123825900130000000
Beale at Market12779183590059661002017510102471783859016500544355772002862361632600000049227179000000000640269250000000024312800161673506445
Broadway St at Battery St678927911900102211000622832260191198792313500570168490032973412140000001697121800000000068543870000000018106003447485007947
California Ave Caltrain Station000000003810002900000000000000000010192400000000600017100000000000000000005700
Castro Street and El Camino Real0000000003000000000014000000000000931340000000007000412000000000000000000000

(省略了 60 行)

我们也可以使用pivot来寻找起点和终点站之间的最短骑行时间。 在这里,pivot已经接受了可选参数Duration,以及min函数,它对每个单元格中的值执行。

commute.pivot('Start Station', 'End Station', 'Duration', min)
End Station2nd at Folsom2nd at South Park2nd at Townsend5th at HowardAdobe on AlmadenArena Green / SAP CenterBeale at MarketBroadway St at Battery StCalifornia Ave Caltrain StationCastro Street and El Camino RealCivic Center BART (7th at Market)Clay at BatteryCommercial at MontgomeryCowper at UniversityDavis at JacksonEmbarcadero at BryantEmbarcadero at FolsomEmbarcadero at SansomeEmbarcadero at VallejoEvelyn Park and RideFranklin at MapleGolden Gate at PolkGrant Avenue at Columbus AvenueHarry Bridges Plaza (Ferry Building)Howard at 2ndJapantownMLK LibraryMarket at 10thMarket at 4thMarket at SansomeMechanics Plaza (Market at Battery)Mezes ParkMountain View Caltrain StationMountain View City HallPalo Alto Caltrain StationPark at OlivePaseo de San AntonioPost at KearnyPowell Street BARTPowell at Post (Union Square)Redwood City Caltrain StationRedwood City Medical CenterRedwood City Public LibraryRengstorff Avenue / California StreetRyland ParkSJSU - San Salvador at 9thSJSU 4th at San CarlosSan Antonio Caltrain StationSan Antonio Shopping CenterSan Francisco Caltrain (Townsend at 4th)San Francisco Caltrain 2 (330 Townsend)San Francisco City HallSan Jose City HallSan Jose Civic CenterSan Jose Diridon Caltrain StationSan Mateo County CenterSan Pedro SquareSan Salvador at 1stSanta Clara County Civic CenterSanta Clara at AlmadenSouth Van Ness at MarketSpear at FolsomSt James ParkStanford in Redwood CitySteuart at MarketTemporary Transbay Terminal (Howard at Beale)Townsend at 7thUniversity and EmersonWashington at KearnyYerba Buena Center of the Arts (3rd @ Howard)
2nd at Folsom61971642680027140700483329306049423926268759900639416282800050623716725000000020826429000000000030030358400000000590208003181494480429165
2nd at South Park616077860078345002901881710357104814903410036927812260004161426168000000602371060000000006366458000000003996300796178027096
2nd at Townsend137676042300311469005465204740436145232509494007735493252210066736726539500000031945539800000000012513374200000000777241002912492590610284
5th at Howard215300384680035753000179412364054341935969560900235474453145002691612503060000002348920200000000025622134700000000375402004552653570553109
Adobe on Almaden00008427500000000000000000007013870000000002290000000441452318000003091461820207358876101003690000000
Arena Green / SAP Center00003056200000000000000000005265460000000004030000000288875685000004404201530166624759116003010000000
Beale at Market219343417387006015500343122153011521617030319800437235149204005352038872000000191316191000000000499395526000000005751730087946190222264
Broadway St at Battery St35142449955500195620052090129070340284128101009611481683570065235121822100000025537631600000000061159979900000000738336001692918850134411
California Ave Caltrain Station00000000821645000628000000000000000000177104841310000000107700087091100000000000000000053100
Castro Street and El Camino Real00000000074000000000499000000000000201108000000000654000953696000000000000000000000

(省略了 60 行)

有人从 Folsom 的二号车站骑到 Beale at Market,距离大约五个街区,非常快(271 秒,约 4.5 分钟)。 2nd Avenue 和 Almaden 的 Adobe 之间没有自行车行程,因为后者在不同的城市。

绘制地图

stations表包含每个自行车站的地理信息,包括纬度,经度和“地标”,它是该站所在城市的名称。

stations = Table.read_table('station.csv')
stations
station_idnamelatlongdockcountlandmarkinstallation
2San Jose Diridon Caltrain Station37.3297-121.90227San Jose8/6/2013
3San Jose Civic Center37.3307-121.88915San Jose8/5/2013
4Santa Clara at Almaden37.334-121.89511San Jose8/6/2013
5Adobe on Almaden37.3314-121.89319San Jose8/5/2013
6San Pedro Square37.3367-121.89415San Jose8/7/2013
7Paseo de San Antonio37.3338-121.88715San Jose8/7/2013
8San Salvador at 1st37.3302-121.88615San Jose8/5/2013
9Japantown37.3487-121.89515San Jose8/5/2013
10San Jose City Hall37.3374-121.88715San Jose8/6/2013
11MLK Library37.3359-121.88619San Jose8/6/2013

(省略了 60 行)

我们可以使用Marker.map_table来绘制一个车站所在位置的地图。 该函数在一个表格上进行操作,该表格的列依次是纬度,经度以及每个点的可选标识符。

Marker.map_table(stations.select('lat', 'long', 'name'))

地图 1

地图使用 OpenStreetMap 创建的,OpenStreetMap 是一个开放的在线地图系统,你可以像使用 Google 地图或任何其他在线地图一样使用。 放大到旧金山,看看车站如何分布。 点击一个标记,看看它是哪个站。

你也可以用彩色圆圈表示地图上的点。 这是旧金山自行车站的地图。

sf = stations.where('landmark', are.equal_to('San Francisco'))
sf_map_data = sf.select('lat', 'long', 'name')
Circle.map_table(sf_map_data, color='green', radius=200)

地图 2

更多信息的地图:join的应用

自行车站位于湾区五个不同的城市。 为了区分每个城市,通过使用不同的颜色,我们首先使用group来标识所有城市,并为每个城市分配一个颜色。

cities = stations.group('landmark').relabeled('landmark', 'city')
cities
citycount
Mountain View7
Palo Alto5
Redwood City7
San Francisco35
San Jose16
colors = cities.with_column('color', make_array('blue', 'red', 'green', 'orange', 'purple'))
colors
citycountcolor
Mountain View7blue
Palo Alto5red
Redwood City7green
San Francisco35orange
San Jose16purple

现在我们可以按照landmark连接stationscolors,之后选取绘制地图所需的列。

joined = stations.join('landmark', colors, 'city')
colored = joined.select('lat', 'long', 'name', 'color')
Marker.map_table(colored)

地图 3

现在五个不同城市由五种不同颜色标记。

要查看大部分自行车租赁的来源,让我们确定起点站:

starts = commute.group('Start Station').sort('count', descending=True)
starts
Start Stationcount
San Francisco Caltrain (Townsend at 4th)25858
San Francisco Caltrain 2 (330 Townsend)21523
Harry Bridges Plaza (Ferry Building)15543
Temporary Transbay Terminal (Howard at Beale)14298
2nd at Townsend13674
Townsend at 7th13579
Steuart at Market13215
Embarcadero at Sansome12842
Market at 10th11523
Market at Sansome11023

(省略了 60 行)

我们可以包含映射这些车站所需的地理数据,首先连接startsstations

station_starts = stations.join('name', starts, 'Start Station')
station_starts
namestation_idlatlongdockcountlandmarkinstallationcount
2nd at Folsom6237.7853-122.39619San Francisco8/22/20137841
2nd at South Park6437.7823-122.39315San Francisco8/22/20139274
2nd at Townsend6137.7805-122.3927San Francisco8/22/201313674
5th at Howard5737.7818-122.40515San Francisco8/21/20137394
Adobe on Almaden537.3314-121.89319San Jose8/5/2013522
Arena Green / SAP Center1437.3327-121.919San Jose8/5/2013590
Beale at Market5637.7923-122.39719San Francisco8/20/20138135
Broadway St at Battery St8237.7985-122.40115San Francisco1/22/20147460
California Ave Caltrain Station3637.4291-122.14315Palo Alto8/14/2013300
Castro Street and El Camino Real3237.386-122.08411Mountain View12/31/20131137

(省略了 58 行)

现在我们只提取绘制地图所需的数据,为每个站添加一个颜色和一个面积。 面积是起始于每个站点的租用次数的 1000 倍,其中选择了常数 1000,以便在地图上以适当的比例绘制圆圈。

starts_map_data = station_starts.select('lat', 'long', 'name').with_columns(
    'color', 'blue',
    'area', station_starts.column('count') * 1000
)
starts_map_data.show(3)
Circle.map_table(starts_map_data)
latlongnamecolorarea
37.7853-122.3962nd at Folsomblue7841000
37.7823-122.3932nd at South Parkblue9274000
37.7805-122.392nd at Townsendblue13674000

(省略了 65 行)

地图 4

旧金山的一大块表明,这个城市的东部是湾区自行车租赁的重点区域。

posted @ 2017-12-27 14:03  绝不原创的飞龙  阅读(54)  评论(0编辑  收藏  举报  来源