feat: add vec_kth_elem function (#5674)

* feat: add vec_kth_elem function

Signed-off-by: pikady <2652917633@qq.com>

* code format

Signed-off-by: pikady <2652917633@qq.com>

* add test sql

Signed-off-by: pikady <2652917633@qq.com>

* change indexing from 1-based to 0-based

Signed-off-by: pikady <2652917633@qq.com>

* improve code formatting and correct spelling errors

Signed-off-by: pikady <2652917633@qq.com>

* Update tests/cases/standalone/common/function/vector/vector.sql

I noticed the two lines are identical. Could you clarify the reason for the change? Thanks!

Co-authored-by: Zhenchi <zhongzc_arch@outlook.com>

---------

Signed-off-by: pikady <2652917633@qq.com>
Co-authored-by: Zhenchi <zhongzc_arch@outlook.com>
This commit is contained in:
pikady
2025-03-18 15:25:53 +08:00
committed by GitHub
parent 1ab4ddab8d
commit eea8b1c730
4 changed files with 250 additions and 1 deletions

View File

@@ -100,6 +100,18 @@ FROM (
SELECT '[7.0, 8.0, 9.0, 10.0]' AS v
) Order By vec_dim(v) ASC;
SELECT vec_kth_elem('[1.0, 2.0, 3.0]', 2);
SELECT v, vec_kth_elem(v, 0) AS first_elem
FROM (
SELECT '[1.0, 2.0, 3.0]' AS v
UNION ALL
SELECT '[4.0, 5.0, 6.0, 7.0]' AS v
UNION ALL
SELECT '[8.0]' AS v
)
WHERE vec_kth_elem(v, 0) > 2.0;
SELECT vec_to_string(vec_subvector('[1.0,2.0,3.0,4.0,5.0]', 0, 3));
SELECT vec_to_string(vec_subvector('[1.0,2.0,3.0,4.0,5.0]', 5, 5));
@@ -121,4 +133,3 @@ FROM (
UNION ALL
SELECT '[4.0, 5.0, 6.0, 10, -8, 100]' AS v
) ORDER BY v;