如何在kotlin中使用map滤波器和reduce函数计算平均值

Question

我想使用kotlin中的map、filter和reduce函数来计算集合中值的平均值。

我以这种方式保存数据：

data class  Track(
val id:String,
var name: String,
var popularity: Int,
var duration: Int,
var explicit: Int,
var artist: String,
var idArtist: String,
var releaseDate: String,
var danceability: Float,
var energy: Float,
var key: Float,
var loudness: Float,
var mode: Float,
var speechness: Float,
var acousticness: Float,
var instrumentalness: Float,
var liveness: Float,
var valence: Float,
var tempo: Float,
var timeSignature: Float

)

我以这种方式按年份对数据进行分组：

fun group_tracks_by_year(tracks: MutableList<Track>): Map<String, List<Track>> {
val grouped_tracks = tracks.groupBy { it.releaseDate.split('-')[0] }.toSortedMap()
return  grouped_tracks
}

例如，我想计算每年的平均节奏，我试图用map函数操作地图集合Map<String, List<Track>>，但我无法操作键值记录。

Answer 1

您不需要filter或reduce。您可以使用sumOf。

您应该使用mapValues，因为map返回的是List，而不是具有相同键的Map。

val tempoAveragesByYear = tracksByYear.mapValues { (year, tracks) -> 
(tracks.sumOf { it.tempo.toDouble() } / tracks.size ).toFloat()
}