python-具有可变循环数的嵌套循环
内容导读
互联网集市收集整理的这篇技术教程文章主要介绍了python-具有可变循环数的嵌套循环,小编现在分享给大家,供广大互联网技能从业者学习和参考。文章包含1784字,纯文字阅读大概需要3分钟。
内容图文
![python-具有可变循环数的嵌套循环](/upload/InfoBanner/zyjiaocheng/669/7c2b639d57cd4134816bf4b7fe1f87f4.jpg)
我有一个在新的基础上输出矩阵的函数.但是,根据矩阵的大小,基础矩阵的数量会有所不同.因此,在简化的“ Matlab伪代码”中:
if matrixsize==1
for a1=1:4
out(a1)=Matrix*basis(a1)
end
elseif matrixsize==2
for a1=1:4
for a2=a1:4
out(a1,a2)=Matrix*basis(a1)*basis(a2)
end
end
elseif matrixsize==3
for a1=1:4
for a2=a1:4
for a3=a2:4
out(a1,a2,a3)=Matrix*basis(a1)*basis(a2)*basis(a3)
end
end
end
elseif ...
等等
是否可以为任何矩阵大小的值编写此代码?
换句话说:是否可以创建一个循环来自动创建上面的循环?
如果这在Matlab中不起作用,那么Python是否有解决方案?
(背景:这个问题来自量子物理学,我想在此以泡利为基础写一个量子态)
这是显示问题的可用Matlab代码:
function T=newbasis(n)
%create a random matrix
m=2^n;
M=randn(m);
%Pauli matrices
s{1}=sparse([1,0;0,1]);
s{2}=sparse([0,1;1,0]);
s{3}=sparse([0,-1i;1i,0]);
s{4}=sparse([1,0;0,-1]);
if n==1
for a1=1:4
T(a1)=trace(M*betterkron(s{a1}));
end
elseif n==2
for a1=1:4
for a2=a1:4
T(a1,a2)=trace(M*betterkron(s{a1},s{a2}));
end
end
elseif n==3
for a1=1:4
for a2=a1:4
for a3=a2:4
T(a1,a2,a3)=trace(M*betterkron(s{a1},s{a2},s{a3}));
end
end
end
else
T=[]
end
%Not very clever but just to keep it simple
function krn=betterkron(A,varargin)
krn = A;
for j = 2:nargin;
krn = kron(krn,varargin{j-1});
end
end
end
解决方法:
尽管原则上可以使用递归函数执行多个这样的循环,但这将很复杂.幸运的是,使用多个循环并不是最好的方法. MATLAB使您可以在N维下标和1维线性索引之间来回转换.因此,您可以对线性索引进行单循环,然后转换回N维下标.所以像这样:
for i=1:numel(Matrix) % loop over linear index
inds = ind2sub(size(Matrix), i); % convert linear index to subscript
% Each index should be greater than or equal to the previous
% e.g. a2=a1:4, a2 starts at a1 so cannot be less than a1
if any(diff(inds) < 0)
continue
end
% Do the calculation
% s{inds} is equivalent to s{i1}, s{i2}, ...
T(i) = trace(M*betterkron(s{inds}));
end
内容总结
以上是互联网集市为您收集整理的python-具有可变循环数的嵌套循环全部内容,希望文章能够帮你解决python-具有可变循环数的嵌套循环所遇到的程序开发问题。 如果觉得互联网集市技术教程内容还不错,欢迎将互联网集市网站推荐给程序员好友。
内容备注
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 gblab@vip.qq.com 举报,一经查实,本站将立刻删除。
内容手机端
扫描二维码推送至手机访问。