np.sqrt详解
时间: 2023-08-27 15:17:23 浏览: 388
np.sqrt() 是 NumPy 库中的一个函数,用于计算给定参数的平方根。
函数定义:
numpy.sqrt(x, /, out=None, *, where=True, casting='same_kind', order='K', dtype=None, subok=True[, signature, extobj])
参数说明:
- x:必需,指定要计算平方根的值。
- out:可选,指定输出结果的位置。
- where:可选,指定元素是否被计算的布尔值数组。
- dtype:可选,指定输出结果的数据类型。
函数返回值:
- 计算给定参数的平方根。
示例:
```python
import numpy as np
# 计算一个数的平方根
a = 16
print(np.sqrt(a)) # 输出:4.0
# 计算一个数组中每个元素的平方根
b = np.array([4, 9, 16, 25])
print(np.sqrt(b)) # 输出:[2. 3. 4. 5.]
```
注意:如果参数为负数,函数将会返回nan(not a number)。
相关问题
详解 psis_all_orth=psis_all_origin[:] for loop_psi in range(len(psis_all_origin)): psi_temp=psis_all_origin[loop_psi] for loop_n in range(loop_psi): psi_temp=psi_temp- np.dot(psis_all_orth[loop_n].conj().T, psis_all_orth[loop_psi])[0][0] * psis_all_orth[loop_n] if np.sqrt(np.dot(psi_temp.conj().T, psi_temp))[0][0] > 0.01: # print('len of psi',loop_psi,np.sqrt(np.dot(psi_temp.conj().T, psi_temp))[0][0]) psi_temp = psi_temp * 1.0 / np.sqrt(np.dot(psi_temp.conj().T, psi_temp)[0][0]) psis_all_orth[loop_psi] = psi_temp
这段代码的作用是对一个包含多个向量的列表进行正交化处理,生成一个新的正交向量列表psis_all_orth。
具体实现方式是:先将原始向量列表psis_all_origin复制一份,赋值给psis_all_orth。然后对于psis_all_origin中的每一个向量,都进行以下操作:
1. 将该向量赋值给临时变量psi_temp。
2. 对于已经处理过的向量列表中的每一个向量,都进行以下操作:
a. 计算该向量与当前处理的向量的内积,将其乘以已处理向量的单位向量。
b. 将该向量减去上述结果,得到一个新的向量。
3. 判断新的向量的模长是否大于0.01,如果是,则将其添加到正交向量列表psis_all_orth中。
最终,psis_all_orth中的向量两两正交,且模长均大于0.01。
阅读全文