old_contents: #!/usr/bin/env python
  # -*- coding: utf-8 -*-
  import numpy as np
  import matplotlib.pyplot as plt
  import matplotlib.patheffects as pe
  
  from vaspy.iter import AniFile
  
  lattice_const = [7.70441, 7.70441, 21]
  orders = [1, 6, 11, 16]
  max_x = 5.5
  
  ani = AniFile("OUT.ANI")
  
  trajs = [[], [], [], []]
  for xyz in ani:
      for j in range(len(orders)):
          data = xyz.data[orders[j]]
          new_data = [0]*3
          for i, c in enumerate(data):
              if i == 0 and c >= max_x:
                  c -= lattice_const[i]
              new_data[i] = c
          trajs[j].append(new_data)
  
  if __name__ == "__main__":
      fig = plt.figure()
      ax = fig.add_subplot(111)
      ax.set_aspect("equal")
      for traj in trajs:
          traj = np.array(traj)
          ax.scatter(traj[:, 0], traj[:, 1], alpha=0.3, facecolor="#93989A", edgecolor="#000000", s=60)
  
      plt.show()
  
  
new_contents: #!/usr/bin/env python
  # -*- coding: utf-8 -*-
  import numpy as np
  import matplotlib.pyplot as plt
  import matplotlib.patheffects as pe
  
  from vaspy.iter import AniFile
  
  lattice_const = [7.70441, 7.70441, 21]
  orders = [1, 6, 11, 16]
  max_x = 5.5
  max_y = max_x
  
  ani = AniFile("OUT.ANI")
  
  trajs = [[], [], [], []]
  for xyz in ani:
      for j in range(len(orders)):
          data = xyz.data[orders[j]]
          new_data = [0]*3
          for i, c in enumerate(data):
              if i == 0 and c >= max_x:
                  c -= lattice_const[i]
              new_data[i] = c
          trajs[j].append(new_data)
  
  if __name__ == "__main__":
      fig = plt.figure()
      ax = fig.add_subplot(111)
      ax.set_aspect("equal")
      for traj in trajs:
          traj = np.array(traj)
          ax.scatter(traj[:, 0], traj[:, 1], alpha=0.3, facecolor="#93989A", edgecolor="#000000", s=60)
  
      ax.set_xlim(0.0, max_x)
      ax.set_ylim(0.0, max_y)
  
      plt.show()
  
  
subject: Add axis limit setting.