2D配列の各列の平均値を計算する方法は？
Nov 30 2020
2D配列の列の平均値を計算しようとしていますが、コードがわかりません。この関数は、各列の平均値を返す必要があります。そして、関数に結果を出力できません。印刷は主な機能である必要があります。
static double average_columns(double matrix[][]) {
int i, j, sum = 0, average=0;
for (i = 0; i < matrix.length; i++) {
for (j = 0; j < matrix[i].length; j++) {
sum=(int) (sum+matrix[i][j]);
}
average=sum/matrix[i].length;
sum=0;
}
return average;
}
回答
1 Som Nov 30 2020 at 12:34
これは、各行と列の合計と平均を計算する方法です。次のサンプルコードは、同じメソッドで合計と平均を出力し、値を返しません。実際の合計と平均を返す必要がある場合は、double []を返す必要があります。double[]には、doubleではなくすべての合計または平均が含まれます。
コード
public class Test {
static int m = 3;
static int n = 3;
public static void main(String[] args) {
int i,j;
int [][]matrix = new int[m][n];
int x = 1; // x fills up the value of the matrix
for (i = 0; i < m; i++)
for (j = 0; j < n; j++)
matrix[i][j] = x++;
System.out.println("The matrix is : \n");
for (i = 0; i < m; i++) {
for (j = 0; j < n; j++) {
System.out.print(matrix[i][j] + " ");
}
System.out.println();
}
System.out.println("\nPrinting the avg of each col ::");
average_columns(matrix);
System.out.println("\nPrinting the avg of each row ::");
average_rows(matrix);
System.out.println("\nPrinting the sum of each col ::");
sum_columns(matrix);
System.out.println("\nPrinting the sum of each row ::");
sum_rows(matrix);
}
public static void average_rows(int matrix[][]) {
int i, j;
double sum = 0, average = 0;
for (i = 0; i < matrix.length; i++) {
for (j = 0; j < matrix[i].length; j++) {
sum=sum+matrix[i][j];
}
average=sum/matrix[i].length;
System.out.println("Average of row " + (i+1) + " = " + average);
sum=0;
}
}
public static void average_columns(int matrix[][]) {
int i, j;
double sum = 0, average = 0;
for (i = 0; i < matrix.length; i++) {
for (j = 0; j < matrix[i].length; j++) {
sum=sum+matrix[j][i];
}
average=sum/matrix[i].length;
System.out.println("Average of column " + (i+1) + " = " + average);
sum=0;
}
}
public static void sum_columns(int matrix[][]) {
int i,j;
double sum = 0;
for (i = 0; i < matrix.length; ++i) {
for (j = 0; j < matrix.length; ++j) {
sum = sum + matrix[j][i];
}
System.out.println("Sum of column " + (i+1) + " = " + sum);
sum = 0;
}
}
public static void sum_rows(int matrix[][]) {
int i,j;
double sum = 0;
for (i = 0; i < matrix.length; ++i) {
for (j = 0; j < matrix.length; ++j) {
sum = sum + matrix[i][j];
}
System.out.println( "Sum of row " + (i+1) + " = " + sum);
sum = 0;
}
}
}
出力
The matrix is :
1 2 3
4 5 6
7 8 9
Printing the avg of each col ::
Average of column 1 = 4.0
Average of column 2 = 5.0
Average of column 3 = 6.0
Printing the avg of each row ::
Average of row 1 = 2.0
Average of row 2 = 5.0
Average of row 3 = 8.0
Printing the sum of each col ::
Sum of column 1 = 12.0
Sum of column 2 = 15.0
Sum of column 3 = 18.0
Printing the sum of each row ::
Sum of row 1 = 6.0
Sum of row 2 = 15.0
Sum of row 3 = 24.0
dreamcrash Nov 30 2020 at 15:41
NxN行列を想定すると、ストリームで解くことができます。
double[][] matrix = { { 1, 5, 15}, { 1, 2, 2}, { 25, 109, 150} };
List<Double> column_average = Arrays.stream(IntStream.range(0, matrix[0].length)
.mapToObj(c1 -> Arrays.stream(matrix).mapToDouble(doubles -> doubles[c1]).toArray())
.toArray(double[][]::new))
.map(i -> Arrays.stream(i).average().getAsDouble())
.collect(Collectors.toList());
以上読みやすい
double[][] matrix_transpose = IntStream.range(0, matrix[0].length)
.mapToObj(c -> Arrays.stream(matrix).mapToDouble(doubles -> doubles[c]).toArray())
.toArray(double[][]::new);
List<Double> column_average = Arrays.stream(matrix_transpose)
.map(col -> Arrays.stream(col).average().getAsDouble())
.collect(Collectors.toList());
行列の転置を行ってから、を使用Arrays.stream(...).average().getAsDouble()して配列の平均を計算します。
ArvindKumarAvinash Nov 30 2020 at 12:13
- 関数からの
double[]代わりにを返す必要がありdoubleます。 - 数値はタイプ、、である
doubleため、タイプは。であるsum必要がありますdouble。
すべての行の長さが等しい場合：
- 各列の値を追加する必要があるため、
matrix[j][i]（の代わりにmatrix[i][j]）をに追加する必要がsumあり、それに応じてにaverage[i]なりますsum / matrix.length。
デモ：
import java.util.Arrays;
public class Main {
public static void main(String[] args) {
// Test
double[][] nums = { { 10, 15, 20 }, { 1, 2, 3 }, { 5, 10, 15 } };
System.out.println(Arrays.toString(averageColumns(nums)));
}
static double[] averageColumns(double matrix[][]) {
int i, j;
double[] average = new double[matrix.length];
for (i = 0; i < matrix.length; i++) {
double sum = 0;
for (j = 0; j < matrix[i].length; j++) {
sum += matrix[j][i];
}
average[i] = sum / matrix.length;
}
return average;
}
}
出力：
[5.333333333333333, 9.0, 12.666666666666666]
行の長さが異なる場合：
- 最初に、のサイズになる行の長さの最大値を見つける必要があり
double[] averageます。 - 最後に、2レベルのネストされたループを使用して、の値を計算します
average[]。外側のループは最大で実行されaverage.length、内側のループは行数まで実行されます。各列の処理中に、カウンター（例int count）を使用して、に追加された値の数を追跡しsumます。内側のループの終わりで、average[i] = sum / count。
デモ：
import java.util.Arrays;
public class Main {
public static void main(String[] args) {
// Test
double[][] nums = { { 10, 15, 20 }, { 1, 2 }, { 5, 10, 15, 25 } };
System.out.println(Arrays.toString(averageColumns(nums)));
}
static double[] averageColumns(double matrix[][]) {
// Find the maximum of the length of rows
int max = matrix[0].length;
for (int i = 0; i < matrix.length; i++) {
if (matrix[i].length > max) {
max = matrix[i].length;
}
}
int i, j;
double[] average = new double[max];
for (i = 0; i < average.length; i++) {
double sum = 0;
int count = 0;
for (j = 0; j < matrix.length; j++) {
if (matrix[j].length - 1 >= i) {
sum += matrix[j][i];
count++;
}
}
average[i] = sum / count;
}
return average;
}
}
出力：
[5.333333333333333, 9.0, 17.5, 25.0]