我想根据这个数组创建一个用 1 或 0 填充的数组
testArray = np.array([7,5,3])
所以最终的结果应该是这样的
[[1,1,1,1,1,1,1],
[1,1,1,1,1],
[1,1,1]]
我想根据这个数组创建一个用 1 或 0 填充的数组
testArray = np.array([7,5,3])
所以最终的结果应该是这样的
[[1,1,1,1,1,1,1],
[1,1,1,1,1],
[1,1,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
.
只需编写一个快速列表理解:
>>> 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)
问题解决了!
numpy 数组中的每一行(和列等)必须具有相同的长度。您可以实现@ChrisWilson4 所做的,并用0
or填充空白部分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 = 0
和background = 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]]
就像@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