ETL table 按变量选择

Question

我正在尝试 table 中的 select 行，并使用 PETL 使用来自原始 table 的信息创建一个新的 table。

我现在的代码是：

import petl as etl


table_all = (
    etl.fromcsv("practice_locations.csv")
        .convert('Practice_Name', 'upper')
        .convert('Suburb', str)
        .convert('State', str)
        .convert('Postcode', int)
        .convert('Lat', str)
        .convert('Long', str)
)


def selection(post_code):
    table_selected = etl.select(table_all, "{Postcode} == 'post_code'")
    print(post_code)
    etl.tojson(table_selected, 'location.json', sort_keys=True)

但我似乎无法按原样使用 selection 函数来填充 table_selected。如果我将 post_code 替换为

，etl.select 调用将起作用

table_selected = etl.select(table_all, "{Postcode} == 4510")

输出正确的 table 显示为：

    +--------------------------------+--------------+-------+----------+--------------+--------------+
    | Practice_Name                  | Suburb       | State | Postcode | Lat          | Long         |
    +================================+==============+=======+==========+==============+==============+
    | 'CABOOLTURE COMBINED PRACTICE' | 'Caboolture' | 'QLD' |     4510 | '-27.085007' | '152.951707' |
    +--------------------------------+--------------+-------+----------+--------------+--------------+

我确定我只是想以错误的方式调用 post_code，但我已经尝试了 PETL 文档中的所有内容，但似乎无法弄清楚。

非常感谢任何帮助。

Answer 1

"{Postcode} == 'post_code'" 不会用传递给您的 selection 函数的值替换 post_code。

您需要格式化 select 字符串（并在使用 format 时转义 {Postcode}）

table_selected = etl.select(table_all, "{{Postcode}} == {post_code}".format(post_code=post_code))

在控制台中测试这个

>>> "{{Postcode}} == {post_code}".format(post_code=1234)
'{Postcode} == 1234'

ETL table 按变量选择

ETL table selection by Variable

python

etl

python-3.x

petl