從Mathematica過渡到Python

Question

我有以下代碼來繪制“共振線”：

TuneDiagram[MyOrder_, MyColor_] := Module[{},
  myLines = 
   Partition[
    Flatten[Table[{{A -> a, B -> +MyOrder - a, C -> p}, {A -> a, 
        B -> -MyOrder + a, C -> p}}, {a, 0, MyOrder}, {p, -MyOrder, 
       MyOrder}]], 3];
  myEquation = A  x + B y == C /. myLines;
  ContourPlot[Evaluate[myEquation], {x, 0, 1}, {y, 0, 1}, 
   ContourStyle -> MyColor, PlotRangePadding -> None, 
   GridLines -> None, 
   ]]

我可以得到像這樣的情節。

我想使用Python / matplotlib獲得相同的結果。

我對這種編程樣式在Python上的“翻譯”感到完全困惑。 任何指針將不勝感激！

Answer 1

根據OP的要求，這里是一種將myLines轉換為參數方程式的方法。 這可能更容易轉換為Python。

pars = {x, y} /. DeleteDuplicates[
   First@Solve[#, First@Variables@*First@#] & /@ myLines];

使用Solve和Variables來獲取函數變量之一中myLines中線性函數的參數方程式。

然后使用ParametricPlot繪制它們。 我猜想Python可以繪制參數圖。

Show[ParametricPlot[#, {x, 0, 1}, {y, 0, 1}] & /@ par,
 PlotRange -> {{0, 1}, {0, 1}}, PlotRangePadding -> None]

希望這可以幫助。