在Python中,.size
函数通常用于获取NumPy数组或Pandas DataFrame等数据结构的大小
import numpy as np
custom_size()
,它接受一个NumPy数组作为输入参数:def custom_size(array):
# 计算数组的形状(行数和列数)
shape = array.shape
# 计算数组的大小
size = np.prod(shape)
return size
# 创建一个示例数组
example_array = np.array([[1, 2, 3], [4, 5, 6]])
# 调用自定义函数并打印结果
print(custom_size(example_array))
这将输出:
6
请注意,这个示例仅适用于NumPy数组。要为其他数据结构(如Pandas DataFrame)编写自定义.size
函数,您需要根据该数据结构的特性进行相应的修改。