maplist = _np.append(maplist, allBOND[allBOND[:, 1] == bead][:, 0])
时间: 2023-06-12 21:04:24 浏览: 49
这是一个Python代码片段,它使用numpy库中的append函数将一个数组(allBOND[allBOND[:, 1] == bead][:, 0])添加到另一个数组(maplist)中。其中allBOND是一个二维数组,它的第二列(allBOND[:, 1])包含一个名为bead的值。[:, 0]表示我们要选择allBOND数组中的第一列数据,并将符合条件的数据添加到maplist数组中。
相关问题
使用C++ eigen库翻译以下python代码import pandas as pd import numpy as np import time import random def main(): eigen_list = [] data = [[1,2,4,7,6,3],[3,20,1,2,5,4],[2,0,1,5,8,6],[5,3,3,6,3,2],[6,0,5,2,19,3],[5,2,4,9,6,3]] g_csi_corr = np.cov(data, rowvar=True) #print(g_csi_corr) eigenvalue, featurevector = np.linalg.eigh(g_csi_corr) print("eigenvalue:",eigenvalue) eigen_list.append(max(eigenvalue)) #以下代码验证求解csi阈值 eigen_list.append(1.22) eigen_list.append(-54.21) eigen_list.append(8.44) eigen_list.append(-27.83) eigen_list.append(33.12) #eigen_list.append(40.29) print(eigen_list) eigen_a1 = np.array(eigen_list) num1 = len(eigen_list) eigen_a2 = eigen_a1.reshape((-1, num1)) eigen_a3 = np.std(eigen_a2, axis=0) eigen_a4 = eigen_a3.tolist() k = (0.016 - 0.014) / (max(eigen_a4) - min(eigen_a4)) eigen_a5 = [0.014 + k * (i - min(eigen_a4)) for i in eigen_a4] tri_threshold = np.mean(eigen_a5)
#include <iostream>
#include <Eigen/Dense>
using namespace Eigen;
int main()
{
std::vector<double> eigen_list;
MatrixXd data(6, 6);
data << 1, 2, 4, 7, 6, 3,
3, 20, 1, 2, 5, 4,
2, 0, 1, 5, 8, 6,
5, 3, 3, 6, 3, 2,
6, 0, 5, 2, 19, 3,
5, 2, 4, 9, 6, 3;
MatrixXd g_csi_corr = data.transpose() * data / 6.0;
EigenSolver<MatrixXd> es(g_csi_corr);
VectorXd eigenvalue = es.eigenvalues().real();
std::cout << "eigenvalue: " << eigenvalue.transpose() << std::endl;
eigen_list.push_back(eigenvalue.maxCoeff());
eigen_list.push_back(1.22);
eigen_list.push_back(-54.21);
eigen_list.push_back(8.44);
eigen_list.push_back(-27.83);
eigen_list.push_back(33.12);
//eigen_list.push_back(40.29);
std::cout << "eigen_list: ";
for (std::vector<double>::iterator it = eigen_list.begin(); it != eigen_list.end(); ++it)
std::cout << *it << " ";
std::cout << std::endl;
int num1 = eigen_list.size();
MatrixXd eigen_a2 = Map<MatrixXd>(eigen_list.data(), num1, 1);
VectorXd eigen_a3 = eigen_a2.array().rowwise().mean().transpose();
VectorXd eigen_a4 = (eigen_a2 - eigen_a3.replicate(num1, 1)).array().abs().rowwise().mean().transpose();
double k = 0.002 / (eigen_a4.maxCoeff() - eigen_a4.minCoeff());
VectorXd eigen_a5 = 0.014 + k * (eigen_a4.array() - eigen_a4.minCoeff());
double tri_threshold = eigen_a5.mean();
std::cout << "tri_threshold: " << tri_threshold << std::endl;
return 0;
}
修正以下代码cntr, u, u0, d, jm, p, fpc = fuzz.cluster.cmeans(train_X.T, 3, 2, error=0.005, maxiter=1000, init=None) train_u, _, _, _, _, _, = fuzz.cluster.cmeans_predict(train_X.T, cntr, 2, error=0.005, maxiter=1000) train_predictions = np.argmax(train_u, axis=0) train_accuracy = accuracy_score(train_y, train_predictions) print(train_y) print(train_predictions) train_y = np.array(train_y) train_u = np.array(train_u) train_auc = roc_auc_score(train_y, train_u.T, multi_class='ovo') loss = np.sum((train_u - train_y.reshape(-1,1)) ** 2) loss_curve.append(loss) accuracy_curve.append(train_accuracy) auc_curve.append(train_auc)
这段代码看起来是使用模糊聚类算法(Fuzzy C-Means)对训练数据进行聚类,并计算训练数据的准确率、AUC以及损失函数。不过代码有一点小问题,应该按照以下方式进行修正:
```python
cntr, u, u0, d, jm, p, fpc = fuzz.cluster.cmeans(train_X.T, 3, 2, error=0.005, maxiter=1000, init=None)
train_u, _, _, _, _, _, = fuzz.cluster.cmeans_predict(train_X.T, cntr, 2, error=0.005, maxiter=1000)
train_predictions = np.argmax(train_u, axis=0)
train_y = np.array(train_y)
train_accuracy = accuracy_score(train_y, train_predictions)
print(train_y)
print(train_predictions)
train_auc = roc_auc_score(train_y, train_u.T, multi_class='ovo')
loss = np.sum((train_u - train_y.reshape(-1,1)) ** 2)
loss_curve.append(loss)
accuracy_curve.append(train_accuracy)
auc_curve.append(train_auc)
```
主要的修改包括:
1. 将 `train_y` 转换为 `numpy` 数组类型。
2. 将计算准确率 `train_accuracy` 的位置调整到 `train_predictions` 后面。
3. 将 `train_auc` 的计算放在 `train_accuracy` 后面,因为需要使用 `train_u`。
4. 将 `loss_curve`、`accuracy_curve` 和 `auc_curve` 的更新放在最后。