Java 类名:com.alibaba.alink.operator.batch.dataproc.format.JsonToVectorBatchOp
Python 类名:JsonToVectorBatchOp
功能介绍
将数据格式从 Json 转成 Vector
要转换为向量,要求Json的key必须可以转换为int,value必须可以转换为数字。否则会报异常。
参数说明
名称 | 中文名称 | 描述 | 类型 | 是否必须? | 取值范围 | 默认值 |
---|---|---|---|---|---|---|
jsonCol | JSON列名 | JSON列的列名 | String | ✓ | 所选列类型为 [STRING] | |
vectorCol | 向量列名 | 向量列对应的列名 | String | ✓ | ||
handleInvalid | 解析异常处理策略 | 解析异常处理策略,可选为ERROR(抛出异常)或者SKIP(输出NULL) | String | “ERROR”, “SKIP” | “ERROR” | |
reservedCols | 算法保留列名 | 算法保留列 | String[] | null | ||
vectorSize | 向量长度 | 向量长度 | Long | -1 |
代码示例
Python 代码
from pyalink.alink import *
import pandas as pd
useLocalEnv(1)
df = pd.DataFrame([
['1', '{"0":1.0,"1":2.0}', '$3$0:1.0 1:2.0', 'f0:1.0,f1:2.0', '1.0,2.0', 1.0, 2.0],
['2', '{"0":4.0,"1":8.0}', '$3$0:4.0 1:8.0', 'f0:4.0,f1:8.0', '4.0,8.0', 4.0, 8.0]])
data = BatchOperator.fromDataframe(df, schemaStr="row string, json string, vec string, kv string, csv string, f0 double, f1 double")
op = JsonToVectorBatchOp()\
.setJsonCol("json")\
.setReservedCols(["row"])\
.setVectorCol("vec")\
.setVectorSize(5)\
.linkFrom(data)
op.print()
Java 代码
import org.apache.flink.types.Row;
import com.alibaba.alink.operator.batch.BatchOperator;
import com.alibaba.alink.operator.batch.dataproc.format.JsonToVectorBatchOp;
import com.alibaba.alink.operator.batch.source.MemSourceBatchOp;
import org.junit.Test;
import java.util.Arrays;
import java.util.List;
public class JsonToVectorBatchOpTest {
@Test
public void testJsonToVectorBatchOp() throws Exception {
List <Row> df = Arrays.asList(
Row.of("1", "{\"0\":\"1.0\",\"1\":\"2.0\"}", "$3$0:1.0 1:2.0", "f0:1.0,f1:2.0", "1.0,2.0", 1.0, 2.0),
Row.of("2", "{\"0\":\"4.0\",\"1\":\"8.0\"}", "$3$0:4.0 1:8.0", "f0:4.0,f1:8.0", "4.0,8.0", 4.0, 8.0)
);
BatchOperator <?> data = new MemSourceBatchOp(df,
"row string, json string, vec string, kv string, csv string, f0 double, f1 double");
BatchOperator <?> op = new JsonToVectorBatchOp()
.setJsonCol("json")
.setReservedCols("row")
.setVectorCol("vec")
.setVectorSize(5)
.linkFrom(data);
op.print();
}
}
运行结果
| row | vec | | —- | —- |
| 1 | $5$1.0 2.0 |
| 2 | $5$4.0 8.0 |