US retailer Wal-Mart has taken a batch of powdered infant formula off the shelf at more than 3,000 of its stores following to death of a newborn in Missouri.