pivot-table - 如何对 DolphinDB 中的表列进行单热编码？

Question

我想对 industryCode 列执行一次性编码，并保留下表中的所有其他列。

t = table(2022.08.01 2022.08.02 2022.08.03 as date, 000001.SZ000002.SZ000003.SZ as windCode, 110111021103 as industryCode)

转换后的表格应该是这样的:

Answer 1

根据您的要求，有两种方法可以解决此问题。

表格:

t = table(2022.08.01 2022.08.02 2022.08.03 as date, `000001.SZ`000002.SZ`000003.SZ as windCode, `1101`1102`1103 as industryCode)

解决方案 #1:

使用内置函数 oneHot(obj, encodingColumns)

oneHot(t, ['industryCode'])

值得一提的是，使用上述方法，您可以一次热编码多个列。

oneHot(t, ['industryCode','windCode'])

解决方案#2:

使用 SQL 查询

res = select iif(isNull(industryCode), 0, 1) from t pivot by date, windCode, industryCode;
nullFill!(res, 0);

我推荐解决方案 #1，因为它的实现要简单得多，而且支持多列。