「 C++ 技巧」“循环移位”代码讲解

一、前言

        小白做神经网络的实验时，遇到矩阵循环填数问题，即矩阵大小不变，旧数据向上移动，新数据插入最后一行。随后做了总结，向上、下、左、右移动的代码都列写在下方，注意需要调用Eigen库。

        附件资料中，有小白学习Eigen库时使用的教程和下载的Eigen库。

---

二、执行代码

/*循环移位
a b 为正数时 向下移动a行 ，向右移动b 列
*/
MatrixXd circshift(MatrixXd data, int A, int B = 0)
{
	UINT row = data.rows();
	UINT col = data.cols();
	MatrixXd out(row, col);
	MatrixXd y(row, col);
	int a = A%row;
	int b = B%col;
	if (a > 0)//向下移动a行
	{
		out.topRows(a) = data.bottomRows(a);
		out.bottomRows(row - a) = data.topRows(row - a);
	}
	else if (a < 0)//向上移动a行
	{
		out.topRows(row + a) = data.bottomRows(row + a);
		out.bottomRows(abs(a)) = data.topRows(abs(a));
	}
	else if (a == 0)
	{
		out.array() = data.array();
	}
 
 
	if (b > 0)//向右移动b列
	{
		y.leftCols(b) = out.rightCols(b);
		y.rightCols(col - b) = out.leftCols(col - b);
	}
	else if (b<0)//向左移动b列
	{
		y.leftCols(col + b) = out.rightCols(col + b);
		y.rightCols(abs(b)) = out.leftCols(abs(b));
 
	}
	else if (b == 0)
	{
		y = out;
	}
	return y;
}

 
	MatrixXd data(3, 3);
	data << 1, 2, 3,
		4, 5, 6,
		7, 8, 9;
	cout << "原始矩阵\n" <<data<< endl;
	cout << "向左移动1次\n" << circshift(data,0, -1) << endl;
	cout << "向右移动2次\n" << circshift(data, 0, 2) << endl;
	cout << "向下移动2次 向左移动1次\n" << circshift(data, 2,-1) << endl;
	cout << "向上移动一次\n" << circshift(data, -1) << endl;