Java 类名:com.alibaba.alink.operator.stream.dataproc.format.VectorToCsvStreamOp
Python 类名:VectorToCsvStreamOp

功能介绍

将数据格式从 Vector 转成 Csv
一条输入对应一条输出结果,输入的vector可以为稀疏格式,也可以为稠密格式。
vector的数据维度不需要保持一致。
setCsvCol设置csv输出列名,setReservedCols设置保留的输入列。
设置SchemaStr时需要注意,字段个数必须小于等于vector列的最小维度。类型为string或者double。
当vector维度大于SchemaStr中的字段个数时,输入vector中后面的维度会被忽略。如果vector在前面维度没有值,输出中也会为空,对应位置会保留。

参数说明

| 名称 | 中文名称 | 描述 | 类型 | 是否必须? | 取值范围 | 默认值 | | —- | —- | —- | —- | —- | —- | —- |

| csvCol | CSV列名 | CSV列的列名 | String | ✓ | | |

| schemaStr | Schema | Schema。格式为”colname coltype[, colname2, coltype2[, …]]”,例如”f0 string, f1 bigint, f2 double” | String | ✓ | | |

| vectorCol | 向量列名 | 向量列对应的列名 | String | ✓ | 所选列类型为 [DENSE_VECTOR, SPARSE_VECTOR, STRING, VECTOR] | |

| csvFieldDelimiter | 字段分隔符 | 字段分隔符 | String | | | “,” |

| handleInvalid | 解析异常处理策略 | 解析异常处理策略,可选为ERROR(抛出异常)或者SKIP(输出NULL) | String | | “ERROR”, “SKIP” | “ERROR” |

| quoteChar | 引号字符 | 引号字符 | Character | | | “”” |

| reservedCols | 算法保留列名 | 算法保留列 | String[] | | | null |

代码示例

Python 代码

  1. from pyalink.alink import *
  2. import pandas as pd
  3. useLocalEnv(1)
  4. df = pd.DataFrame([
  5. ['1', '{"f0":"1.0","f1":"2.0"}', '$4$0:1.0 3:2.0', 'f0:1.0,f1:2.0', '1.0,2.0', 1.0, 2.0],
  6. ['2', '{"f0":"4.0","f1":"8.0"}', '$3$0:4.0 2:8.0', 'f0:4.0,f1:8.0', '4.0,8.0', 4.0, 8.0]])
  7. data = StreamOperator.fromDataframe(df, schemaStr="row string, json string, vec string, kv string, csv string, f0 double, f1 double")
  8. op = VectorToCsvStreamOp()\
  9. .setVectorCol("vec")\
  10. .setReservedCols(["row"])\
  11. .setCsvCol("csv")\
  12. .setSchemaStr("f0 double, f1 double, f2 double")\
  13. .linkFrom(data)
  14. op.print()
  15. StreamOperator.execute()

Java 代码

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

运行结果

| row | csv | | —- | —- |

| 2 | 4.0,,8.0 |

| 1 | 1.0,, |