如何为MLDataTable中的列定义/更改MLDataValue.ValueType

Question

我正在从给定的.csv文件加载function('date_part', 'dow', event.date)。根据输入文件的内容自动推断每一列的数据类型。
稍后处理表时，我需要可预测的显式类型。

如何在加载文件时强制执行某种类型，或者在第二步中更改类型？

简化示例：

MLDataTable

Answer 1

使用MLDataColumn的{​​{1}}方法从具有所需基础类型的现有列中派生新列：

map(to:)

产生以下输出：

let squaresArrayInt = (1...5).map{$0 * $0}
var table = try! MLDataTable(dictionary: ["Ints" :  squaresArrayInt])
print(table)

let squaresColumnDouble = table["Ints"].map(to: Double.self)
table.addColumn(squaresColumnDouble, named: "Doubles")
print(table)