如何将Snowflake SQL的数组索引转换为SQLAlchemy Core代码？

你不能直接用Python的[0]语法访问func.split()的结果，因为func.split()返回的是SQLAlchemy的表达式对象，不是Python列表。针对Snowflake的场景，有两种简单的解决方式：

方法一：用Snowflake的ELEMENT_AT函数替代数组索引 #

Snowflake中array[N]的语法等价于ELEMENT_AT(array, N)，你可以在SQLAlchemy里直接调用这个函数来实现数组元素的获取：

from sqlalchemy import func

# 对应原SQL的 TRIM(SPLIT(CATEGORIES, ',')[0])
# 注意：Snowflake默认数组索引是1-based，若原SQL用[0]需确认是否开启了0-based索引
main_category = func.trim(func.element_at(func.split(CATEGORIES, ','), 1))
# 若你的Snowflake环境设置了0-based索引，第二个参数传0即可
# main_category = func.trim(func.element_at(func.split(CATEGORIES, ','), 0))

方法二：用SPLIT_PART函数直接取分割元素 #

如果只是取分割后的第一个元素，用Snowflake的SPLIT_PART会更直接，不需要处理数组：

from sqlalchemy import func

# 效果和原SQL完全一致，取第一个分割元素并去除首尾空格
main_category = func.trim(func.split_part(CATEGORIES, ',', 1))

注意事项 #

Snowflake默认数组索引是1-based，所以原SQL中的[0]在默认配置下会返回NULL，除非你已经执行过SET ARRAY_INDEXING_START = 0;修改了索引起始规则，使用时要注意匹配你的环境配置。

内容的提问来源于stack exchange，提问作者Bryley