2

我想根据这个数组创建一个用 1 或 0 填充的数组

testArray = np.array([7,5,3])

所以最终的结果应该是这样的

[[1,1,1,1,1,1,1],
 [1,1,1,1,1],
 [1,1,1]]
4

4 回答 4

1

这将为您提供一个参差不齐的 dtype 数组object

>>> result = np.array([np.ones(a) for a in testArray])
>>> print result
[[ 1.  1.  1.  1.  1.  1.  1.] [ 1.  1.  1.  1.  1.] [ 1.  1.  1.]]

对于零,只需使用np.zeros.

于 2013-05-03T14:51:01.580 回答
1

只需编写一个快速列表理解:

>>> holder = [np.ones((testArray[i])) for i in range(len(testArray))]
>>> holder
[array([[ 1.,  1.,  1.,  1.,  1.,  1.,  1.]]), array([[ 1.,  1.,  1.,  1.,  1.]]), array([[ 1.,  1.,  1.]])]

如果您希望它采用写入格式,您可以随时重塑它:

>>> np.array(holder).reshape(3,1)
array([[array([ 1.,  1.,  1.,  1.,  1.,  1.,  1.])],
       [array([ 1.,  1.,  1.,  1.,  1.])],
       [array([ 1.,  1.,  1.])]], dtype=object)

问题解决了!

于 2013-05-03T17:29:52.243 回答
1

numpy 数组中的每一行(和列等)必须具有相同的长度。您可以实现@ChrisWilson4 所做的,并用0or填充空白部分np.nan。创建一个空数组,其行数等于长度lengths,列数等于最大行:

fill = 1    # or `0` or `np.nan`
background = 0 # or `np.nan`
lengths = np.array([7,5,3])

a = np.ones((lengths.size, lengths.max()))*background

并用你的fill价值填充它:

for row, length in enumerate(lengths):
    a[row,:length] = fill

a
#array([[ 1.,  1.,  1.,  1.,  1.,  1.,  1.],
#       [ 1.,  1.,  1.,  1.,  1.,  0.,  0.],
#       [ 1.,  1.,  1.,  0.,  0.,  0.,  0.]])

或者,对于fill = 0background = np.nan

array([[  0.,   0.,   0.,   0.,   0.,   0.,   0.],
       [  0.,   0.,   0.,   0.,   0.,  nan,  nan],
       [  0.,   0.,   0.,  nan,  nan,  nan,  nan]])

或者,您可以使用纯 Python 方式(不使用 numpy)制作列表列表,如下所示:

fill = 1
lengths = [7,5,3]
a = [ [fill]*length for length in lengths ]

a
#[[1, 1, 1, 1, 1, 1, 1], [1, 1, 1, 1, 1], [1, 1, 1]]
于 2013-05-03T14:45:28.010 回答
-1

就像@JoshAdel 在评论中所说的那样, int 数组不能是锯齿状的,这意味着行不能是不同的长度。这是你要找的吗?

public class soArray {
public static void main(String[] args) {

    int[][] testArray = soArray.array(7,5,3);
    
    for (int i = 0; i < testArray.length; i++){
        for (int j = 0; j < testArray[0].length; j++){
            System.out.print(testArray[i][j]);
        }
        System.out.println();
    }
}
public static int[][] array(int a, int b, int c){

    int max;
    if(a > b && a > c)
        max = a;
    else if(b > a && b > c)
        max = b;
    else
        max = c;
        
    int[][] out = new int[3][max];
    
    for (int i = 0; i < max; i++){
        if(i < a)
            out[0][i] = 1;
        else
            out[0][i] = 0;
    }
    for(int i = 0; i< b; i++){
        if(i < b)
            out[1][i] = 1;
        else
            out[1][i] = 0;
    }
    for(int i = 0; i < c; i++){
        if(i < c)
            out[2][i] = 1;
        else
            out[2][i] = 0;
    }
    
    return out;
}

}

它会打印出来:

1111111

1111100

1110000

于 2013-05-03T14:24:11.053 回答