val numCols = 4
val threshold = 10000
val arr1 = Array.ofDim[Int](numCols)
df.map {
r =>(0 until numCols).map(i => r.getInt(i))
}.collect()(0).copyToArray(arr1).map(_ > threshold)
方法2:
val arr2 = df.collect()(0).getValuesMap[Int](df.columns)
.values.toArray.map(_ > threshold)
1条答案
按热度按时间khbbv19g1#
有两种方法可以实现这一点,至少可以避免中间转换为字符串。假设
df
是Dataframe:方法1:
方法2: