Java 类名：com.alibaba.alink.operator.batch.dataproc.format.VectorToKvBatchOp
Python 类名：VectorToKvBatchOp

功能介绍

将数据格式从 Vector 转成 Key-value，批式组件
一条输入数据对应一条输出数据，setKvCol设置kv输出列名，setReservedCols设置保留的输入列。
在输出kv列中，vector的下标对应key，vector的值对应的value。

参数说明

名称 中文名称 描述 类型 是否必须？ 取值范围 默认值
kvCol KV列名 KV列的列名 String ✓
vectorCol 向量列名 向量列对应的列名 String ✓ 所选列类型为 [DENSE_VECTOR, SPARSE_VECTOR, STRING, VECTOR]
handleInvalid 解析异常处理策略 解析异常处理策略，可选为ERROR（抛出异常）或者SKIP（输出NULL） String “ERROR”, “SKIP” “ERROR”
kvColDelimiter 分隔符 当输入数据为稀疏格式时，key-value对之间的分隔符 String “,”
kvValDelimiter 分隔符 当输入数据为稀疏格式时，key和value的分割符 String “:”
reservedCols 算法保留列名 算法保留列 String[] null

代码示例

Python 代码

  1. from pyalink.alink import *
  3. import pandas as pd
  5. useLocalEnv(1)
  7. df = pd.DataFrame([
  8.     ['1', '{"f0":"1.0","f1":"2.0"}', '$3$0:1.0 1:2.0', 'f0:1.0,f1:2.0', '1.0,2.0', 1.0, 2.0],
  9.     ['2', '{"f0":"4.0","f1":"8.0"}', '$3$0:4.0 1:8.0', 'f0:4.0,f1:8.0', '4.0,8.0', 4.0, 8.0]])
  11. data = BatchOperator.fromDataframe(df, schemaStr="row string, json string, vec string, kv string, csv string, f0 double, f1 double")
  13. op = VectorToKvBatchOp()\
  14.     .setVectorCol("vec")\
  15.     .setReservedCols(["row"])\
  16.     .setKvCol("kv")\
  17.     .linkFrom(data)
  19. op.print()

Java 代码

  1. import org.apache.flink.types.Row;
  3. import com.alibaba.alink.operator.batch.BatchOperator;
  4. import com.alibaba.alink.operator.batch.dataproc.format.VectorToKvBatchOp;
  5. import com.alibaba.alink.operator.batch.source.MemSourceBatchOp;
  6. import org.junit.Test;
  8. import java.util.Arrays;
  9. import java.util.List;
  11. public class VectorToKvBatchOpTest {
  12.     @Test
  13.     public void testVectorToKvBatchOp() throws Exception {
  14.         List <Row> df = Arrays.asList(
  15.             Row.of("1", "{\"f0\":\"1.0\",\"f1\":\"2.0\"}", "$3$0:1.0 1:2.0", "f0:1.0,f1:2.0", "1.0,2.0", 1.0, 2.0),
  16.             Row.of("2", "{\"f0\":\"1.0\",\"f1\":\"2.0\"}", "$3$0:4.0 1:8.0", "f0:1.0,f1:2.0", "1.0,2.0", 1.0, 2.0)
  17.         );
  18.         BatchOperator <?> data = new MemSourceBatchOp(df,
  19.             "row string, json string, vec string, kv string, csv string, f0 double, f1 double");
  20.         BatchOperator <?> op = new VectorToKvBatchOp()
  21.             .setVectorCol("vec")
  22.             .setReservedCols("row")
  23.             .setKvCol("kv")
  24.             .linkFrom(data);
  25.         op.print();
  26.     }
  27. }

运行结果

| row | kv | | —- | —- |

| 1 | 0:1.0,1:2.0 |

| 2 | 0:4.0,1:8.0 |