Question

我有一个将一些数据加载到2D numpy数组中的函数。我想让函数调用指定可以从开头和结尾删除的行数和列数。如果未指定参数，则返回所有数据。

import numpy as np
function load_data(min_row, max_row, min_col, max_col):
    a = np.loadtxt('/path/to/mydata.txt')[min_row:max_row,min_col:max_col]

现在，min_row和min_col可以默认为0。如何设置max_col和max_row的默认值以引用数组的结尾？

我唯一的解决方案是：

function load_data(min_row=0, max_row=None, min_col=0, max_col=None):
    a = np.loadtxt('/path/to/mydata.txt')
    if not max_row: max_row = a.shape[0]
    if not max_col: max_col = a.shape[1]
    a = a[min_row:max_row,min_col:max_col]

是否有更好的解决方案，例如：

function load_data(min_row=0, max_row="end", min_col=0, max_col="end"):
    a = np.loadtxt('/path/to/mydata.txt')[min_row:max_row,min_col:max_col]

对于记录，示例数据可以是：

np.array([[1, 2, 3, 4, 5],
          [1, 2, 3, 4, 5],
          [1, 2, 3, 4, 5],
          [1, 2, 3, 4, 5],
          [1, 2, 3, 4, 5]])

Answer 1

您可以直接在切片中使用None，例如

x = np.arange(10)
x[2:None]    # array([5, 6, 7, 8, 9])

或者你可以写下你的功能：

function load_data(min_row=0, max_row=None, min_col=0, max_col=None):
    a = np.loadtxt('/path/to/mydata.txt')
    a = a[min_row:max_row,min_col:max_col]

在这里，您也可以用None替换最小默认值。这是有效的，因为None用作slice object中的默认值。有关在numpy切片中使用None的更明确的文档，请参阅Basic Slicing docs description末尾的“注释”框。

numpy数组结束的索引

1 个答案: