PyODPS还提供了IPython的插件,来更方便得操作ODPS。

IPython增强

首先,针对命令行增强,也有相应的命令。让我们先加载插件:

%load_ext odps
%enter
<odps.inter.Room at 0x11341df10>

此时全局会包含o和odps变量,即ODPS入口。

o.get_table('dual')
odps.get_table('dual')
odps.Table
  name: odps_test_sqltask_finance.`dual`
  schema:
    c_int_a                 : bigint
    c_int_b                 : bigint
    c_double_a              : double
    c_double_b              : double
    c_string_a              : string
    c_string_b              : string
    c_bool_a                : boolean
    c_bool_b                : boolean
    c_datetime_a            : datetime
    c_datetime_b            : datetime
%stores
default desc
name
iris 安德森鸢尾花卉数据集

对象名补全

PyODPS拓展了IPython原有的代码补全功能,支持在书写o.get_xxx 这样的语句时,自动补全对象名。

例如,在IPython中输入下列语句(<tab>不是实际输入的字符,而是当所有输入完成后,将光标移动到相应位置, 并按 Tab 键):

o.get_table(<tab>

如果已知需要补全对象的前缀,也可以使用

o.get_table('tabl<tab>

IPython会自动补全前缀为tabl的表。

对象名补全也支持补全不同Project下的对象名。下列用法都被支持:

o.get_table(project='project_name', name='tabl<tab>
o.get_table('tabl<tab>', project='project_name')

如果匹配的对象有多个,IPython会给出一个列表,其最大长度由 options.completion_size 给出, 默认为10。

SQL命令

PyODPS还提供了SQL插件,来执行ODPS SQL。下面是单行SQL:

%sql select * from pyodps_iris limit 5
sepallength sepalwidth petallength petalwidth name
0 5.1 3.5 1.4 0.2 Iris-setosa
1 4.9 3.0 1.4 0.2 Iris-setosa
2 4.7 3.2 1.3 0.2 Iris-setosa
3 4.6 3.1 1.5 0.2 Iris-setosa
4 5.0 3.6 1.4 0.2 Iris-setosa

多行SQL可以使用%%sql的命令

%%sql

select * from pyodps_iris
where sepallength < 5
limit 5
sepallength sepalwidth petallength petalwidth name
0 4.9 3.0 1.4 0.2 Iris-setosa
1 4.7 3.2 1.3 0.2 Iris-setosa
2 4.6 3.1 1.5 0.2 Iris-setosa
3 4.6 3.4 1.4 0.3 Iris-setosa
4 4.4 2.9 1.4 0.2 Iris-setosa

如果想执行参数化SQL查询,则需要替换的参数可以使用:参数的方式。

In [1]: %load_ext odps

In [2]: mytable = 'dual'

In [3]: %sql select * from :mytable
|==========================================|   1 /  1  (100.00%)         2s
Out[3]:
   c_int_a  c_int_b  c_double_a  c_double_b  c_string_a  c_string_b c_bool_a  \
0        0        0       -1203           0           0       -1203     True

  c_bool_b         c_datetime_a         c_datetime_b
0    False  2012-03-30 23:59:58  2012-03-30 23:59:59

设置SQL运行时参数,可以通过%set设置到全局,或者在sql的cell里用SET进行局部设置。

In [17]: %%sql
         set odps.sql.mapper.split.size = 16;
         select * from pyodps_iris;

这个会局部设置,不会影响全局的配置。

In [18]: %set odps.sql.mapper.split.size = 16

这样设置后,后续运行的SQL都会使用这个设置。

持久化pandas DataFrame到ODPS表

PyODPS还提供把pandas DataFrame上传到ODPS表的命令:

import pandas as pd
import numpy as np

df = pd.DataFrame(np.arange(9).reshape(3, 3), columns=list('abc'))
%persist df pyodps_pandas_df

这里的第0个参数df是前面的变量名,pyodps_pandas_df是ODPS表名。