如何创建类似numpy nd数组数据结构

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

import pandas as pd

data = {'col1': [1, 2, 3], 'col2': [4, 5, 6]}
df = pd.DataFrame(data)

class NDArray:
    def __init__(self, shape):
        self.shape = shape
        self.data = [0] * (shape[0] * shape[1])  # 初始化数据为0
    
    def get(self, row, col):
        index = row * self.shape[1] + col
        return self.data[index]
    
    def set(self, row, col, value):
        index = row * self.shape[1] + col
        self.data[index] = value

# 创建一个2x3的二维数组
array = NDArray((2, 3))
array.set(0, 0, 1)
array.set(0, 1, 2)
array.set(0, 2, 3)
array.set(1, 0, 4)
array.set(1, 1, 5)
array.set(1, 2, 6)