spy.cpp 600 B

1234567891011121314151617181920212223242526272829
  1. #import <iostream>
  2. #import <vector>
  3. #import "../matplotlibcpp.h"
  4. namespace plt = matplotlibcpp;
  5. int main()
  6. {
  7. const int n = 20;
  8. std::vector<std::vector<double>> matrix;
  9. for (int i = 0; i < n; ++i) {
  10. std::vector<double> row;
  11. for (int j = 0; j < n; ++j) {
  12. if (i == j)
  13. row.push_back(-2);
  14. else if (j == i - 1 || j == i + 1)
  15. row.push_back(1);
  16. else
  17. row.push_back(0);
  18. }
  19. matrix.push_back(row);
  20. }
  21. plt::spy(matrix, 5, {{"marker", "o"}});
  22. plt::show();
  23. return 0;
  24. }