将二维数组转换成一维数组

可以使用flatten()函数将二维数组转换成一维数组。例如：

arr = [[1, 2], [3, 4], [5, 6]]
new_arr = [item for sublist in arr for item in sublist]
print(new_arr)

输出：

[1, 2, 3, 4, 5, 6]

也可以使用numpy库中的ravel()函数将二维数组转换成一维数组。例如：

import numpy as np

arr = np.array([[1, 2], [3, 4], [5, 6]])
new_arr = arr.ravel()
print(new_arr)

输出：

[1 2 3 4 5 6]

相关问题

如何将二维数组转化成一维数组 C++

在C++中，将二维数组转换为一维数组通常涉及到深度复制或者扁平化的过程。你可以通过遍历二维数组并逐个添加元素到新的一维数组中来实现。这里是一个简单的示例：

#include <vector>

std::vector<int> flatten(const std::vector<std::vector<int>>& two_dim_array) {
    std::vector<int> one_dim_array;
    for (const auto& row : two_dim_array) {
        one_dim_array.insert(one_dim_array.end(), row.begin(), row.end());
    }
    return one_dim_array;
}

在这个例子中，`flatten`函数接受一个二维整数向量`two_dim_array`作为输入，它创建了一个新的`one_dim_array`来存储转换后的数据。使用`insert`函数将每行元素追加到一维数组的末尾。

如果你有一个静态大小的二维数组`int arr[n][m]`，也可以这样做：

std::vector<int> flatten(int n, int m, int(&arr)[n][m]) {
    std::vector<int> one_dim(n * m);
    int index = 0;
    for (int i = 0; i < n; ++i)
        for (int j = 0; j < m; ++j)
            one_dim[index++] = arr[i][j];
    return one_dim;
}

这里假设数组的大小在运行时已知。

怎么把二维数组转换成一维数组

在 NumPy 中，我们可以使用`flatten()`函数将一个二维数组转换为一维数组。`flatten()`函数会将二维数组中的所有元素按照行优先的顺序展开成一个一维数组。

例如，假设我们有一个二维数组`arr`，其中包含了 3 行 4 列共 12 个元素：

arr = [[1, 2, 3, 4],
       [5, 6, 7, 8],
       [9, 10, 11, 12]]

我们可以使用以下代码将其转换为一维数组：

import numpy as np

arr_1d = np.array(arr).flatten()

这样会返回一个新的一维数组`arr_1d`，其中包含了原来二维数组中的所有元素。需要注意的是，`flatten()`函数返回的是一个拷贝，不会改变原始的二维数组`arr`。

除了`flatten()`函数，我们还可以使用`ravel()`函数将二维数组转换为一维数组。`ravel()`函数与`flatten()`函数类似，但是返回的是原始数组的一个视图，而不是拷贝。因此，如果对返回的一维数组进行修改，会同时影响原始的二维数组。

arr_1d = np.ravel(arr)