如何将第一列设置为空 np.zeros numPy 矩阵的常量值?
How to set first column to a constant value of an empty np.zeros numPy matrix?
我正在为水 table 模型设置一些边界条件,我能够将整个第一行设置为常数值,但不能将整个第一列设置为常数值。我正在使用 np.zeros((11, 1001))
来制作一个空矩阵。有谁知道为什么我成功地定义了第一行,而不是第一列?我已经注意到下面有问题的行。
import numpy as np
x = range(0, 110, 10)
time = range(0, 5005, 5)
xSize = len(x)
timeSize = len(time)
dx = 10
dt = 5
Sy = 0.1
k = 0.002
head = np.zeros((11, 1001))
head[0:][0] = 16 # sets the first row to 16
head[0][0:] = 16 # DOESN'T set the first column to 16
for t in time:
for i in x[1:len(x)-1]:
head[t+1][i] = head[t][i] + ((dt*k)/(2*Sy)) * (((head[t][i-1]**2) - (2*head[t][i]**2) + (head[t][i+1]**2)) / (dx**2))
您可以使用类似的语法。
In [12]: head = np.zeros((11,101))
In [13]: head
Out[13]:
array([[ 0., 0., 0., ..., 0., 0., 0.],
[ 0., 0., 0., ..., 0., 0., 0.],
[ 0., 0., 0., ..., 0., 0., 0.],
...,
[ 0., 0., 0., ..., 0., 0., 0.],
[ 0., 0., 0., ..., 0., 0., 0.],
[ 0., 0., 0., ..., 0., 0., 0.]])
In [14]: head[:,0] = 42.0
In [15]: head
Out[15]:
array([[ 42., 0., 0., ..., 0., 0., 0.],
[ 42., 0., 0., ..., 0., 0., 0.],
[ 42., 0., 0., ..., 0., 0., 0.],
...,
[ 42., 0., 0., ..., 0., 0., 0.],
[ 42., 0., 0., ..., 0., 0., 0.],
[ 42., 0., 0., ..., 0., 0., 0.]])
你所要做的就是改变
head[0][0:]
到
head[:, 0] = 16
如果你想改变第一行,你可以这样做:
head[0, :] = 16
编辑:
以防万一你也想知道如何更改任意数量的值 row/column:
myArray = np.zeros((6, 6))
现在我们将第 2 行的 3 个值设置为 16:
myArray[2, 1:4] = 16.
array([[ 0., 0., 0., 0., 0., 0.],
[ 0., 0., 0., 0., 0., 0.],
[ 0., 16., 16., 16., 0., 0.],
[ 0., 0., 0., 0., 0., 0.],
[ 0., 0., 0., 0., 0., 0.],
[ 0., 0., 0., 0., 0., 0.]])
同样适用于列:
myArray[2:5, 4] = -4.
array([[ 0., 0., 0., 0., 0., 0.],
[ 0., 0., 0., 0., 0., 0.],
[ 0., 16., 16., 16., -4., 0.],
[ 0., 0., 0., 0., -4., 0.],
[ 0., 0., 0., 0., -4., 0.],
[ 0., 0., 0., 0., 0., 0.]])
如果您还想更改某些值,例如同时有两个不同的行,你可以这样做:
myArray[[0, 5], 0:3] = 10.
array([[ 10., 10., 10., 0., 0., 0.],
[ 0., 0., 0., 0., 0., 0.],
[ 0., 16., 16., 16., -4., 0.],
[ 0., 0., 0., 0., -4., 0.],
[ 0., 0., 0., 0., -4., 0.],
[ 10., 10., 10., 0., 0., 0.]])
我正在为水 table 模型设置一些边界条件,我能够将整个第一行设置为常数值,但不能将整个第一列设置为常数值。我正在使用 np.zeros((11, 1001))
来制作一个空矩阵。有谁知道为什么我成功地定义了第一行,而不是第一列?我已经注意到下面有问题的行。
import numpy as np
x = range(0, 110, 10)
time = range(0, 5005, 5)
xSize = len(x)
timeSize = len(time)
dx = 10
dt = 5
Sy = 0.1
k = 0.002
head = np.zeros((11, 1001))
head[0:][0] = 16 # sets the first row to 16
head[0][0:] = 16 # DOESN'T set the first column to 16
for t in time:
for i in x[1:len(x)-1]:
head[t+1][i] = head[t][i] + ((dt*k)/(2*Sy)) * (((head[t][i-1]**2) - (2*head[t][i]**2) + (head[t][i+1]**2)) / (dx**2))
您可以使用类似的语法。
In [12]: head = np.zeros((11,101))
In [13]: head
Out[13]:
array([[ 0., 0., 0., ..., 0., 0., 0.],
[ 0., 0., 0., ..., 0., 0., 0.],
[ 0., 0., 0., ..., 0., 0., 0.],
...,
[ 0., 0., 0., ..., 0., 0., 0.],
[ 0., 0., 0., ..., 0., 0., 0.],
[ 0., 0., 0., ..., 0., 0., 0.]])
In [14]: head[:,0] = 42.0
In [15]: head
Out[15]:
array([[ 42., 0., 0., ..., 0., 0., 0.],
[ 42., 0., 0., ..., 0., 0., 0.],
[ 42., 0., 0., ..., 0., 0., 0.],
...,
[ 42., 0., 0., ..., 0., 0., 0.],
[ 42., 0., 0., ..., 0., 0., 0.],
[ 42., 0., 0., ..., 0., 0., 0.]])
你所要做的就是改变
head[0][0:]
到
head[:, 0] = 16
如果你想改变第一行,你可以这样做:
head[0, :] = 16
编辑:
以防万一你也想知道如何更改任意数量的值 row/column:
myArray = np.zeros((6, 6))
现在我们将第 2 行的 3 个值设置为 16:
myArray[2, 1:4] = 16.
array([[ 0., 0., 0., 0., 0., 0.],
[ 0., 0., 0., 0., 0., 0.],
[ 0., 16., 16., 16., 0., 0.],
[ 0., 0., 0., 0., 0., 0.],
[ 0., 0., 0., 0., 0., 0.],
[ 0., 0., 0., 0., 0., 0.]])
同样适用于列:
myArray[2:5, 4] = -4.
array([[ 0., 0., 0., 0., 0., 0.],
[ 0., 0., 0., 0., 0., 0.],
[ 0., 16., 16., 16., -4., 0.],
[ 0., 0., 0., 0., -4., 0.],
[ 0., 0., 0., 0., -4., 0.],
[ 0., 0., 0., 0., 0., 0.]])
如果您还想更改某些值,例如同时有两个不同的行,你可以这样做:
myArray[[0, 5], 0:3] = 10.
array([[ 10., 10., 10., 0., 0., 0.],
[ 0., 0., 0., 0., 0., 0.],
[ 0., 16., 16., 16., -4., 0.],
[ 0., 0., 0., 0., -4., 0.],
[ 0., 0., 0., 0., -4., 0.],
[ 10., 10., 10., 0., 0., 0.]])