numpy.ndarray.flatten()函數

該函數主要用來快速扁平化數組，請看如下代碼：

import numpy as np


class Debug:
    def __init__(self):
        self.array1 = np.array([[1, 2], [3, 4]])
        self.array2 = np.ones((2, 2, 2))

    def mainProgram(self):
        print("The value of array1 is: ")
        print(self.array1)
        print("The value of flattened array is: ")
        array2 = self.array1.flatten()
        print(array2)
        print("The value of array2 is: ")
        print(self.array2)
        print("The value of flattened array is: ")
        print(self.array2.flatten())


if __name__ == '__main__':
    main = Debug()
    main.mainProgram()
"""
The value of array1 is: 
[[1 2]
 [3 4]]
The value of flattened array is: 
[1 2 3 4]
The value of array2 is: 
[[[1. 1.]
  [1. 1.]]

 [[1. 1.]
  [1. 1.]]]
The value of flattened array is: 
[1. 1. 1. 1. 1. 1. 1. 1.]
"""