If it's  a permanent or semi permanent color  it's OK to go swim in a water or pool that has chlorine for the disinfecting purposes. If you used a temporary hair die you should be aware  that every time you wash your hair it will fade the color. See hair color direction on the box.